- add Piezometer data via json (Gschliegraben)

gschliefgraben_glasfaser/models.py

@@ -8,25 +8,83 @@ Python version: 3.10
 from datetime import datetime
 # from config import db, ma
 
-# import os
+import os
 from sqlalchemy import (Column, Integer,
-                        String, DateTime, ForeignKey, Numeric)
+                        String, DateTime, ForeignKey, Numeric, SmallInteger, create_engine)
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import session, relationship
 #from marshmallow import Schema
-from marshmallow_sqlalchemy import SQLAlchemyAutoSchema
-from db.pg_models import create_pg_session
-
+from marshmallow_sqlalchemy import SQLAlchemySchema, SQLAlchemyAutoSchema
+from marshmallow import fields
+# from db.pg_models import create_pg_session
+from sqlalchemy import func, desc, asc
+import sqlalchemy.orm.session
+from sqlalchemy.orm import sessionmaker, relationship
 
 Base = declarative_base()
+def new_id_factory():
+    pg_session = create_pg_session()
+    _MYTABLE_ID_ = pg_session.query(func.max(Observation.id)).scalar()
+    _MYTABLE_ID_ += 1
+    return _MYTABLE_ID_
 
+def create_pg_session() -> sqlalchemy.orm.sessionmaker:
+    """ create postgres db session """
+    dbschema = ''
+    db_user = os.environ.get("POSTGIS_DBUSER")
+    db_password = os.environ.get("POSTGIS_DBPASSWORD")
+    db_url = os.environ.get("POSTGIS_DBURL")
+    engine = create_engine(
+        "postgresql+psycopg2://" + db_user + ":" + db_password + "@" + db_url,
+        connect_args={'options': '-csearch_path={}'.format(dbschema)},
+        isolation_level="READ UNCOMMITTED")
+    session_maker = sessionmaker(bind=engine)
+    session = session_maker()
 
+    # Base.metadata.create_all(engine)
+    return session
+
+class Dataset(Base):
+    """ dataset class """
+    __tablename__ = 'dataset'
+    __table_args__ = {"schema": "gba"}
+
+    id = Column('dataset_id', Integer, primary_key=True)
+    name = Column('name', String)
+    is_published = Column('is_published', SmallInteger)
+    is_hidden = Column('is_hidden', SmallInteger)
+    dataset_type = Column('dataset_type', String)
+    observation_type = Column('observation_type', String)
+    value_type = Column('value_type', String)
+
+    last_time = Column('last_time', DateTime)
+    last_value = Column('last_value', Numeric(20, 10))
+    fk_last_observation_id = Column(
+        'fk_last_observation_id',
+        Integer
+    )
+    # last_observation = relationship(
+    #     "Observation", foreign_keys=[fk_last_observation_id])
+
+    first_time = Column('first_time', DateTime)
+    first_value = Column('first_value', Numeric(20, 10))
+    fk_first_observation_id = Column(
+        'fk_first_observation_id',
+        Integer
+    )
+    # first_observation = relationship("Observation", foreign_keys=[
+    #                                  fk_first_observation_id])
+
+    observations = relationship(
+        'Observation', back_populates='dataset', lazy=True)
+    
 class Observation(Base):
     """ observation class """
     __tablename__ = 'observation'
     __table_args__ = {"schema": "gba"}
 
-    id = Column('observation_id', Integer, primary_key=True)
+    id = Column('observation_id', Integer,
+                primary_key=True)
     name = Column('name', String)
     value_type = Column('value_type', String)
     # pitch = Column('PITCH', String)
@@ -37,16 +95,20 @@ class Observation(Base):
     sta_identifier = Column('sta_identifier', String)
     value_quantity = Column('value_quantity', Numeric(20, 10), nullable=False)
 
-    # fk_dataset_id = Column('fk_dataset_id', Integer,
-    #                        ForeignKey('gba.dataset.dataset_id'))
-    # dataset = relationship("Dataset", lazy="joined",
-    #                        foreign_keys=[fk_dataset_id])
     fk_dataset_id = Column(Integer, ForeignKey(
         'gba.dataset.dataset_id'), nullable=False)
-    # dataset = relationship("Dataset", back_populates="observations")
+    dataset = relationship("Dataset", back_populates="observations")
 
-class ObservationSchema(SQLAlchemyAutoSchema):
+
+class ObservationSchema(SQLAlchemySchema):
     """ Platform class """
+    DateTime = fields.DateTime(attribute='result_time')  # Or vice-versa
+    # value_quantity = fields.Integer(attribute='Value')
+    # id = fields.Integer(attribute='id')
+    Value = fields.Integer(attribute='value_quantity')
+    id = fields.Integer(attribute='sta_identifier')
+    # sta_identifier= fields.String(default=uuid.uuid4()),
+
     class Meta:
         """ Platform class """
         model = Observation
@@ -72,7 +134,7 @@ class Person(Base):
             self.login, self.lname)
 
 
-class PersonSchema(SQLAlchemyAutoSchema):
+class PersonSchema(SQLAlchemyAutoSchema):    
     """ Platform class """
     class Meta:
         """ Platform class """
@@ -81,3 +143,26 @@ class PersonSchema(SQLAlchemyAutoSchema):
         load_instance = True
         #pg_session: session = create_pg_session()
         sqla_session: session = create_pg_session()
+
+def create_db():
+    # db_url  = 'sqlite:///db.sqlite'    
+    # engine = create_engine(db_url, echo = True )  
+    # Base.metadata.drop_all(bind=engine)
+    # Base.metadata.create_all(engine)
+    
+    """ create postgres db session """
+    dbschema = ''
+    db_user = os.environ.get("POSTGIS_DBUSER")
+    db_password = os.environ.get("POSTGIS_DBPASSWORD")
+    db_url = os.environ.get("POSTGIS_DBURL")
+    engine = create_engine(
+        "postgresql+psycopg2://" + db_user + ":" + db_password + "@" + db_url,
+        connect_args={'options': '-csearch_path={}'.format(dbschema)},
+        isolation_level="READ UNCOMMITTED", echo = True)
+    # session_maker = sessionmaker(bind=engine)
+    # session = session_maker()
+    Base.metadata.drop_all(bind=engine)
+    Base.metadata.create_all(engine)
+        
+if __name__ == "__main__":
+    create_db()