add observation class for importing observations

This commit is contained in:
Arno Kaimbacher 2022-02-23 16:46:47 +01:00
parent 1dafc5824c
commit 7dd739b04e
5 changed files with 75 additions and 28 deletions

View file

@ -10,10 +10,10 @@ import os
# currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
# parentdir = os.path.dirname(currentdir)
# sys.path.insert(0, parentdir)
from db.pg_models import create_pg_session
from sqlalchemy.orm import session
from gschliefgraben_glasfaser.models import PersonSchema
from models import Person, PersonSchema
from models import ObservationSchema, Person, PersonSchema, Observation
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
# print(response.json()) # shows the response's JSON response body, if it has one
@ -21,21 +21,29 @@ from models import Person, PersonSchema
def main():
# db_user = os.environ.get("POSTGIS_DBUSER")
# print(db_user)
''' main method '''
db_user = os.environ.get("POSTGIS_DBUSER")
print(db_user)
pg_session: session = create_pg_session()
pg_person: Person = pg_session.query(Person).first()
# pg_person: Person = pg_session.query(Person).first()
observation: Observation = pg_session.query(Observation).first()
print (observation)
# serialize db data to json
person_schema = PersonSchema()
dump_data = person_schema.dump(pg_person)
# person_schema = PersonSchema()
# dump_data = person_schema.dump(pg_person)
# print(dump_data)
# serialize db data to json
observation_schema = ObservationSchema()
dump_data = observation_schema.dump(observation)
print(dump_data)
# deserialize
load_data: Person = person_schema.load(dump_data)
print(load_data)
create(dump_data)
# # deserialize
# load_data: Person = person_schema.load(dump_data)
# print(load_data)
# create(dump_data)
def create(person_json: PersonSchema):
@ -48,14 +56,14 @@ def create(person_json: PersonSchema):
login = person_json.get('login')
#lname = person.get('lname')
session = create_pg_session()
db_session = create_pg_session()
# existing_person = Person.query \
# .filter(Person.login == login) \
# .one_or_none()
existing_person: bool = ( \
session.query(Person) \
.filter(Person.login == login)
db_session.query(Person) \
.filter(Person.login == login)
.one_or_none()
)
@ -66,9 +74,9 @@ def create(person_json: PersonSchema):
# deserialize to object
new_person: Person = schema.load(person_json)
# Add the person to the database
session.add(new_person)
session.commit()
# Add the person to the database
db_session.add(new_person)
db_session.commit()
# Serialize and return the newly created person in the response
data = schema.dump(new_person)