from PyQt5.QtCore import QVariant
from qgis._core import QgsCoordinateReferenceSystem, QgsGeometry, QgsPointXY
from qgis.core import QgsFields, QgsFeature, QgsField

from .po_waterlevels import PoWaterlevels


class PoWaterlevelsQgs(PoWaterlevels):

    def __init__(self):
        super().__init__()
        self.fields = None
        self.crs = QgsCoordinateReferenceSystem(4326, QgsCoordinateReferenceSystem.EpsgCrsId)

    def getWaterlevelsFeatures(self):
        print("getWaterlevelsFeatures: Erzeuge Features...")

        self.fields = QgsFields()
        self.fields.append(QgsField('shortname', QVariant.String))
        self.fields.append(QgsField('number', QVariant.LongLong))
        self.fields.append(QgsField('agency', QVariant.String))
        self.fields.append(QgsField('timestamp', QVariant.DateTime))
        self.fields.append(QgsField('value', QVariant.Double))
        self.fields.append(QgsField('stateMnwMhw', QVariant.String))
        self.fields.append(QgsField('stateNswHsw', QVariant.String))

        features = []
        waterlevels = self.getWaterlevels()
        if waterlevels is None or len(waterlevels) == 0:
            print("getWaterlevelsFeatures: Fehler: Keine Pegelstände erhalten")
            return None

        for station in waterlevels:
            feature = self._getFeatureForStation(station)
            if feature is not None:
                features.append(feature)

        print("getWaterlevelsFeatures: %d Features erzeugt" % (len(features),))
        return features

    def _getFeatureForStation(self, station):
        # noinspection DuplicatedCode
        if station['geometry']['longitude'] is None or station['geometry']['latitude'] is None:
            print("_getFeatureForStation: WARN: Station hat fehlende Koordinaten: %s" % (station['attributes']['shortname'],))
            return None

        feature = QgsFeature(self.fields)

        longitude = station['geometry']['longitude']
        latitude = station['geometry']['latitude']
        feature.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(longitude, latitude)))

        feature.setAttribute('shortname', station['attributes']['shortname'])
        feature.setAttribute('number', station['attributes']['number'])
        feature.setAttribute('agency', station['attributes']['agency'])
        feature.setAttribute('timestamp', station['attributes']['timestamp'])
        feature.setAttribute('value', station['attributes']['value'])
        feature.setAttribute('stateMnwMhw', station['attributes']['stateMnwMhw'])
        feature.setAttribute('stateNswHsw', station['attributes']['stateNswHsw'])

        return feature