This commit is contained in:
yumoqing 2025-10-19 17:36:10 +08:00
parent 42d799bb49
commit 6c88be2071

View File

@ -487,13 +487,16 @@ class SQLor(object):
if sqlstring is None: if sqlstring is None:
return [] return []
recs = [] recs = []
await self.execute(sqlstring,{},lambda x:recs.append(x)) async for r in self.execute(sqlstring, {}):
recs.append(r)
return recs return recs
async def fields(self,tablename=None): async def fields(self,tablename=None):
sqlstring = self.fieldsSQL(tablename) sql = self.fieldsSQL(tablename)
recs = [] recs = []
await self.execute(sqlstring,{},lambda x:recs.append(x)) async for r in self.execute(sql, {}):
recs.append(r)
ret = [] ret = []
for r in recs: for r in recs:
r.update({'type':self.db2modelTypeMapping.get(r['type'].lower(),'unknown')}) r.update({'type':self.db2modelTypeMapping.get(r['type'].lower(),'unknown')})
@ -504,14 +507,15 @@ class SQLor(object):
async def primary(self,tablename): async def primary(self,tablename):
sqlstring = self.pkSQL(tablename) sqlstring = self.pkSQL(tablename)
recs = [] recs = []
await self.execute(sqlstring,{},lambda x:recs.append(x)) async for r in self._get_data(sqlstring,{}):
# print('sql=', sqlstring, 'recs=', recs) recs.append(recs)
return recs return recs
async def fkeys(self,tablename): async def fkeys(self,tablename):
sqlstring = self.fkSQL(tablename) sqlstring = self.fkSQL(tablename)
recs = [] recs = []
await self.execute(sqlstring,{},lambda x:recs.append(x)) async for r in self._get_data(sqlstring, {}):
recs.append(r)
return recs return recs
async def createTable(self,tabledesc): async def createTable(self,tabledesc):
@ -551,14 +555,14 @@ class SQLor(object):
return desc return desc
async def rollback(self): async def rollback(self):
if self.async_mode: if inspect.iscoroutinefunction(self.conn.rollback):
await self.conn.rollback() await self.conn.rollback()
else: else:
self.conn.rollback() self.conn.rollback()
self.dataChanged = False self.dataChanged = False
async def commit(self): async def commit(self):
if self.async_mode: if inspect.iscoroutinefunction(self.conn.commit):
await self.conn.commit() await self.conn.commit()
else: else:
self.conn.commit() self.conn.commit()