geomon/gschliefgraben_glasfaser/main.py

'''
Tutorial link: https://realpython.com/flask-connexion-rest-api-part-2/
https://github.com/realpython/materials/blob/master/flask-connexion-rest-part-2/version_1/people.py
Sqlalchemy version: 1.2.15
Python version: 3.7
'''

import os
# import sys, inspect
# 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
# 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
# print(response.content)  # get the data content of the response


def main():
    # db_user = os.environ.get("POSTGIS_DBUSER")
    # print(db_user)

    pg_session: session = create_pg_session()
    pg_person: Person = pg_session.query(Person).first()

    # serialize db data to json
    person_schema = PersonSchema()
    dump_data = person_schema.dump(pg_person)
    print(dump_data)

    # deserialize
    load_data: Person = person_schema.load(dump_data)
    print(load_data)
    create(dump_data)


def create(person_json: PersonSchema):
    """
    This function creates a new person in the people structure
    based on the passed-in person data
    :param person:  person to create in people structure
    :return:        201 on success, 406 on person exists
    """

    login = person_json.get('login')
    #lname = person.get('lname')
    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)        
        .one_or_none()
    )

    # Can we insert this person?
    if existing_person is None:
         # Create a person instance using the schema and the passed in person
        schema = PersonSchema()
        # deserialize to object
        new_person: Person = schema.load(person_json)

        # Add the person to the database        
        session.add(new_person)
        session.commit()

        # Serialize and return the newly created person in the response
        data = schema.dump(new_person)
        return data, 201
    # Otherwise, nope, person exists already
    else:
        print(409, f'Person {login} exists already')


if __name__ == "__main__":
    main()