Lei PAN
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
from builtins import list
|
|
|
|
import sqlalchemy
|
|
from sqlalchemy import MetaData, Column, String, Integer, DateTime, Table
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy_utils import database_exists, create_database
|
|
|
|
import params
|
|
from pojo.ReserveResultPojo import ReserveResultPojo
|
|
from pojo.captcha_error_contact_pojo import ContactInErrorPojo, ERROR_TYPE_CAPTCHA
|
|
|
|
|
|
class LocalDbManager:
|
|
|
|
def __init__(self, path: str):
|
|
self.session = Session(self.init_db(path))
|
|
|
|
def init_db(self, path: str):
|
|
uri_for_db = "sqlite:///{}/{}.db?check_same_thread=false".format(path, "appointment")
|
|
print(uri_for_db)
|
|
# 2.-Turn on database engine
|
|
db_engine = sqlalchemy.create_engine(uri_for_db) # ensure this is the correct path for the sqlite file.
|
|
if not database_exists(uri_for_db):
|
|
create_database(uri_for_db)
|
|
connextion = db_engine.connect()
|
|
if not db_engine.dialect.has_table(connextion,
|
|
ContactInErrorPojo.__tablename__): # If table don't exist, Create.
|
|
metadata = MetaData(db_engine)
|
|
# Create a table with the appropriate Columns
|
|
Table(ContactInErrorPojo.__tablename__, metadata,
|
|
Column('id', Integer, primary_key=True, autoincrement=True),
|
|
Column('mail', String),
|
|
Column('phone', String),
|
|
Column('passport', String),
|
|
Column('last_name', String),
|
|
Column('first_name', String),
|
|
Column('ccid', String),
|
|
Column('position', Integer),
|
|
Column('error_type', Integer),
|
|
Column('update_at', DateTime),
|
|
Column('create_at', DateTime))
|
|
# Implement the creation
|
|
metadata.create_all()
|
|
|
|
return db_engine
|
|
|
|
def insert_or_update(self, instance: ContactInErrorPojo):
|
|
self.session.merge(instance)
|
|
self.session.commit()
|
|
|
|
def handle_success(self, reservePojo: ReserveResultPojo):
|
|
# delete the contact from table
|
|
self.session.query(ContactInErrorPojo).filter(ContactInErrorPojo.mail == reservePojo.email).delete()
|
|
|
|
|