def __init__(self, table_name, s_model=None, t_model=None, m_model=None): self.table_name = table_name if s_model: self.s_api = DbApi(get_source_db(), s_model) if t_model: self.t_api = DbApi(get_target_db(), t_model) if m_model: self.m_api = DbApi(get_migration_db(), m_model)
class MigrationClient(object): s_id = app.config['SYNC_LAST_PK'] t_id = None m_id = None s_data = None t_data = None m_data = None latest_time = None limit_num = app.config['SYNC_LIMIT_NUM'] def __init__(self, table_name, s_model=None, t_model=None, m_model=None): self.table_name = table_name if s_model: self.s_api = DbApi(get_source_db(), s_model) if t_model: self.t_api = DbApi(get_target_db(), t_model) if m_model: self.m_api = DbApi(get_migration_db(), m_model) def t_create(self): self.t_id = self.t_api.add(self.t_data) def t_update(self): self.t_api.edit(self.t_id, self.t_data) def m_detail(self): self.m_data = self.m_api.get_row( table_name=self.table_name, pk_source=self.s_id, ) if self.m_data: self.m_id = self.m_data.id def m_create(self): current_time = datetime.utcnow() m_data = { 'table_name': self.table_name, 'pk_source': self.s_id, 'pk_target': self.t_id, 'latest_time': self.latest_time, 'create_time': current_time, 'update_time': current_time, } self.m_id = self.m_api.add(m_data) def m_update(self): current_time = datetime.utcnow() m_data = { 'table_name': self.table_name, 'pk_source': self.s_id, 'pk_target': self.t_id, 'latest_time': self.latest_time, # 'create_time': current_time, 'update_time': current_time, } self.m_api.edit(self.m_id, m_data)
def test03(): """独立连接""" print(test03.__doc__) target_user_api = DbApi(get_target_db(), User) print(target_user_api.count()) target_user_api = DbApi(get_target_db(), User) print(target_user_api.count())
def test01(): """共用连接""" print(test01.__doc__) target_user_api = DbApi(target_db, User) print(target_user_api.count()) target_user_api = DbApi(target_db, User) print(target_user_api.count())
def test02(): """独立连接""" print(test02.__doc__) source_user_api = DbApi(get_source_db(), EAPUser) print(source_user_api.count()) source_user_api = DbApi(get_source_db(), EAPUser) print(source_user_api.count())
def test01(): """共用连接""" print(test01.__doc__) source_user_api = DbApi(source_db, EAPUser) print(source_user_api.count()) source_user_api = DbApi(source_db, EAPUser) print(source_user_api.count())
def test04(): """独立连接""" print(test04.__doc__) print(DbApi(get_target_db(), User).count()) print(DbApi(get_target_db(), User).count())
def test02(): """共用连接""" print(test02.__doc__) print(DbApi(target_db, User).count()) print(DbApi(target_db, User).count())