在操作數據庫的業務里，我們系統采用了orm框架 ,避免了過多的寫sql，利用實體對數據庫進行操作
需求： 賬戶系統里的account表是進行了分表，分表規則為accountid進行20取模，測試環境分為多套環境
a,b,c環境，需要在對數據庫操作時區分環境
1、使用sqlalchemy
base = declarative_base()
def get_model(name,env):
if. env =='a':
engine = create_engine('mysql+pymysql.....這里是a環境的配置')
elif env =='b':
.......
base.metadata.reflect(engine)
table = base.metadata.table[name]
mapper(t, table)
Base.metadata.clear()
return t
當然這里的參數name是需要額外做處理的，因為存在分表的規則，所以需要在先得到具體的表名再找到對應的model
2、使用flask_sqlalchemy
通過元類編程找出對應的model
class Account(object):
__mapper = {}
@staticmethod
def model( account_id, env = 'c'):
if env == 'c':
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:root@xxxx'
else :
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:root@xxx'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
db =SQLAlchemy(app)
table_index = account_id%20
class_name='account_%d' % table_index
ModelClass = Account.__mapper.get(class_name, None)
if ModelClass is None:
ModelClass = type(class_name, (db.Model,),
{
'__module__':__name__,
'__name__':class_name,
'__tablename__':'account_%d' % table_index,
'account_id':db.Column(db.Integer, primary_key=True),
'main_id':db.Column(db.Integer)
})
Account.__mapper[class_name]=ModelClass
cls = ModelClass()
cls.account_id = account_id
return cls