Python 对于分表的操作

在操作数据库的业务里，我们系统采用了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

巴特西

Python 对于分表的操作

最新文章

热门文章