test_framework
/
AutomationTest
mirror of https://github.com/yanchunhuo/AutomationTest.git
4
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

增强base db

This commit is contained in:
yanchunhuo 2022-08-04 13:09:14 +08:00
parent 815ed90553
commit f2ab1dc54d
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
common_projects/sqliteOpt/base_db.py
View File

@ -3,7 +3,7 @@
# @author yanchunhuo
# @description
# @created 2022-08-04T09:52:09.525Z+08:00
# @last-modified 2022-08-04T10:34:56.676Z+08:00
# @last-modified 2022-08-04T13:09:00.283Z+08:00
# github https://github.com/yanchunhuo
class Base_DB:
@ -17,8 +17,12 @@ class Base_DB:
result_object=self.db_session.query(self.model).filter_by(**attrs).first()
return result_object
def filter_objects(self,obj:object):
def filter_objects(self,obj:object,page_index:int=None,page_size:int=None,):
attrs=obj.__dict__
attrs.pop('_sa_instance_state')
result_objects=self.db_session.query(self.model).filter_by(**attrs).all()
if page_index and page_size:
result_objects=self.db_session.query(self.model).filter_by(**attrs).\
limit(page_size).offset((page_index-1)*page_size).all()
else:
result_objects=self.db_session.query(self.model).filter_by(**attrs).all()
return result_objects