- check if observation already exits in Db, then insert

gschliefgraben_glasfaser/main.py

@@ -6,16 +6,19 @@ Python version: 3.7
 '''
 
 import os
+from tokenize import String
+import uuid
 # import sys, inspect
 # currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
 # parentdir = os.path.dirname(currentdir)
 # sys.path.insert(0, parentdir)
 # import requests
 from sqlalchemy.orm import session
-
-from gschliefgraben_glasfaser.models import ObservationSchema, Person, PersonSchema, Observation
+from sqlalchemy import func
+from gschliefgraben_glasfaser.models import ObservationSchema, Person, PersonSchema, Observation, create_pg_session
 from gschliefgraben_glasfaser.my_api import MyApi
-from db.pg_models import create_pg_session
+from datetime import datetime, date, timedelta
+# from db.pg_models import create_pg_session
 #from models import Person, PersonSchema
 # response = requests.get('https://api.com/')
 # print(response)  # shows the response's HTTP status code
@@ -47,8 +50,8 @@ def main():
     # print(load_data)
 
     # request ortmann api
-    # token = 'eyJraWQiOiJlakFmX1MwMTBMU3doS0Zod05wZDQtQkZPYTM4cDRYRE1zU1hFa0lrRlhFIiwiYWxnIjoiUlMyNTYifQ.eyJ2ZXIiOjEsImp0aSI6IkFULkZRUHNCOWh5Snd6eEM5d3ZWelRvaTNpZVlMWlJiT3U4YzFCbWJWRGM1SFkiLCJpc3MiOiJodHRwczovL2Rldi01MjUwMDA2Lm9rdGEuY29tL29hdXRoMi9kZWZhdWx0IiwiYXVkIjoiYXBpOi8vZGVmYXVsdCIsImlhdCI6MTY0NTc4Mjg0NSwiZXhwIjoxNjQ1Nzg2NDQ1LCJjaWQiOiIwb2EyOWhzdGZ3RnFya1BrUDVkNyIsInNjcCI6WyJnc2NobGllZmdyYWJlbiJdLCJzdWIiOiIwb2EyOWhzdGZ3RnFya1BrUDVkNyJ9.c-pTs-3VJMnFO2SOqxOvsABAloprUmOjk6SO9J71NrgLj7claKZOMLZxRyUeSBLWCJFFNI3A6xMd4twEexjJdUR8UEM4U50srxr2p_enaMm1_jZTSt_76u6H05kwV-A2AOQPkx-Fxxaj_PDjT7w43Zlg6SUEoT11uGKR6KtxVYbclGtWgOR7wvH4NZav-P_EDjHwHxbk2kQSf7tBU1JbWl74Xt58gzv1t8VNtLYLICabRsuTNQUNiO7Y1rtUEav4ugf7WZMIY1cP_4rCupZrAFbxrnyprAuXA2x01Z9hbFmiaK0QDlrwHcCHL_1fKvj9uIbO5JeI1x81X6g7eAxQdA'
-    # response = requests.get('https://api.dgnss-sensors.com/gschliefgraben?sensors=("inclino1_14")',
+    # response =
+    # requests.get('https://api.dgnss-sensors.com/gschliefgraben?sensors=("inclino1_14")',
     #                         headers={
     #                             'Authorization': 'Bearer' + token,
     #                             'cache-control': 'no-cache',
@@ -57,16 +60,116 @@ def main():
     #                         },
     #                         data='grant_type=client_credentials&scope=gschliefgraben')
     # print(response)
+
+    # The size of each step in days
+
+    # consider the start date as 2021-february 1 st
+    start_date = date(2021, 2, 28)
+    # consider the end date as 2021-march 1 st
+    end_date = date(2022, 3, 1)
+
+    # delta time
+    delta = timedelta(days=1)
     token_api = os.environ.get("TOKEN_API")
     test_api = MyApi(token_api)
-    data = test_api.getSensorData("inclino1_14")
-    observation_array = (data['FeatureCollection']['Features'][0]['geometry']['properties'][0])
-    print(observation_array)
-    # create(dump_data)
-    # # deserialize to db model
-    observation_schema = ObservationSchema(many=True)
-    observations: Observation = observation_schema.load(observation_array)
-    print(observations)
+
+    # iterate over range of dates
+    while start_date <= end_date:
+        # print(start_date, end="\n")
+        query_date = start_date.strftime('%Y-%m-%d')
+        create_db_observations(query_date, test_api, pg_session)
+        start_date += delta
+
+    # for i in rrule(DAILY , dtstart=start_date,until=end_date):
+    #     print(i.strftime('%Y%b%d'),sep='\n')
+
+
+    # query_date = "2022-02-28"
+    # create_db_observations(query_date, test_api, pg_session)
+    # query_date_obj = datetime.strptime(query_date, "%Y-%m-%d")
+    # data = test_api.getSensorData("inclino1_14", query_date)
+    # observation_array = (data['FeatureCollection']
+    #                      ['Features'][0]['geometry']['properties'][0])
+    # print(observation_array)
+
+
+
+    # max_id = pg_session.query(func.max(Observation.id)).scalar()
+    # if max_id is None:
+    #     max_id = -1
+    # # pg_session.bulk_save_objects(observations)
+    # for observation_json in observation_array:
+    #     ob_date_time = observation_json.get('DateTime')      
+    #     datetime_obj = datetime.strptime(ob_date_time, "%Y-%m-%dT%H:%M:%S.%fZ")
+    #     if datetime_obj.date() != query_date_obj.date():
+    #         continue
+    #     max_id = max_id + 1
+    #     create_observation(observation_json, pg_session, max_id)
+
+    # pg_session.commit()
+
+def create_db_observations(query_date, test_api, pg_session):
+    ''' to do '''
+    query_date_obj = datetime.strptime(query_date, "%Y-%m-%d")
+    data = test_api.getSensorData("inclino1_14", query_date)
+    observation_array = (data['FeatureCollection']
+                         ['Features'][0]['geometry']['properties'][0])
+    # print(observation_array)
+
+    max_id = pg_session.query(func.max(Observation.id)).scalar()
+    if max_id is None:
+        max_id = -1
+    # pg_session.bulk_save_objects(observations)
+    for observation_json in observation_array:
+        ob_date_time = observation_json.get('DateTime')      
+        datetime_obj = datetime.strptime(ob_date_time, "%Y-%m-%dT%H:%M:%S.%fZ")
+        if datetime_obj.date() != query_date_obj.date():
+            continue
+        ob_value = observation_json.get('Value') 
+        if ob_value is None:
+            continue
+        max_id = max_id + 1
+        create_observation(observation_json, pg_session, max_id)
+    pg_session.commit()
+    print("observations for date " +query_date+  "succesfully imported \n")
+
+
+def create_observation(observation_json: ObservationSchema, db_session, max_id):
+    """
+    This function creates a new observation in the people structure
+    based on the passed-in observation data
+    :param observation:  person to create in people structure
+    :return:        201 on success, observation on person exists
+    """
+
+    ob_id = observation_json.get('id')
+    # db_session = create_pg_session()
+
+    existing_observation: bool = (
+        db_session.query(Observation)
+        .filter(Observation.id == ob_id)
+        .one_or_none()
+    )
+
+    # Can we insert this observation?
+    if existing_observation is None:
+        # Create a person instance using the schema and the passed in person
+        schema = ObservationSchema()
+        # deserialize to object
+        new_observation: Observation = schema.load(observation_json)
+        new_observation.id = max_id + 1
+        new_observation.sta_identifier = str(uuid.uuid4())
+
+        # Add the person to the database
+        db_session.add(new_observation)
+        # db_session.commit()
+
+        # Serialize and return the newly created person in the response
+        data = schema.dump(new_observation)
+        return data, 201
+    # Otherwise, nope, person exists already
+    else:
+        print(409, f'Observation {ob_id} exists already')
 
 
 def create(person_json: PersonSchema):