Base.py 586 B

1234567891011121314151617181920212223
  1. from sqlalchemy.orm import sessionmaker
  2. from sqlalchemy import create_engine,text
  3. class Base():
  4. __engine = None
  5. def __init__(self, db):
  6. if Base.__engine is None:
  7. Base.__engine = create_engine('mysql+pymysql://{user}:{password}@{host}:{port}/{dbname}'.format(**db))
  8. def execute(self, query):
  9. with Base.__engine.connect() as conn:
  10. conn.execute(text(query))
  11. conn.commit()
  12. @staticmethod
  13. def get_session():
  14. DBSession = sessionmaker(bind=Base.__engine)
  15. session = DBSession()
  16. return session