sql_models.py 1.4 KB

12345678910111213141516171819202122232425262728293031
  1. from sqlalchemy import create_engine, MetaData
  2. from sqlalchemy.orm import sessionmaker, scoped_session
  3. from config import using_config
  4. import urllib
  5. class DB():
  6. def __init__(self, config=None):
  7. self.config = using_config if not config else config
  8. self.engine = self.DBengine()
  9. self.session = self.DBSession()
  10. self.metadata = MetaData(self.engine)
  11. def DBengine(self):
  12. # 初始化数据库连接:
  13. db_uri = 'mysql+pymysql://' \
  14. '{username}:{password}@{host}:{port}/{database}'.format(username=urllib.parse.quote(self.config['username']),
  15. password=urllib.parse.quote(self.config['password']),
  16. host=self.config['host'],
  17. port=self.config['port'],
  18. database=self.config['database'])
  19. engine = create_engine(db_uri, pool_size=1024, pool_recycle=1800,
  20. pool_pre_ping=True, max_overflow=100, echo=False)
  21. return engine
  22. def DBSession(self):
  23. # 创建DBSession类型:
  24. session_factory = sessionmaker(bind=self.engine)
  25. DBSession = scoped_session(session_factory)
  26. DBSession = DBSession()
  27. return DBSession