git.datanom.net - pwp.git/blame_incremental

... / ...

Commit	Line	Data
	1	import logging
	2	logging.basicConfig(level=logging.INFO)
	3	logger = logging.getLogger(__name__)
	4
	5	class Base(object):
	6
	7	def query(self, sql, placeholders):
	8	res = None
	9	cur = self.conn.cursor()
	10	try:
	11	cur.execute(sql, placeholders)
	12	res = cur.fetchall()
	13	except Exception as e:
	14	logger.error("{0}".format(e))
	15	pass
	16	finally:
	17	cur.close()
	18
	19	return res
	20
	21	def update(self, sql, placeholders):
	22	rows = None
	23	cur = self.conn.cursor()
	24	try:
	25	cur.execute(sql, placeholders)
	26	self.conn.commit()
	27	rows = cur.rowcount
	28	except Exception as e:
	29	self.conn.rollback()
	30	raise Exception(e)
	31	finally:
	32	cur.close()
	33
	34	return rows
	35
	36	def insert(self, sql, placeholders):
	37	lastid = None
	38	cur = self.conn.cursor()
	39	try:
	40	cur.execute(sql, placeholders)
	41	if self.__class__.__name__ == 'Postgres':
	42	cur.execute('select lastval()')
	43	lastid = cur.fetchone()[0]
	44	else:
	45	lastid = cur.lastrowid
	46	self.conn.commit()
	47	except Exception as e:
	48	self.conn.rollback()
	49	raise Exception(e)
	50	finally:
	51	cur.close()
	52
	53	return lastid
	54
	55	def ddl(self, sql):
	56	cur = self.conn.cursor()
	57	try:
	58	if self.__class__.__name__ == 'Mysql':
	59	for r in cur.execute(sql, multi=True):
	60	pass
	61	else:
	62	cur.execute(sql)
	63	self.conn.commit()
	64	except Exception as e:
	65	self.conn.rollback()
	66	raise Exception(e)
	67	finally:
	68	cur.close()
	69
	70	def __del__(self):
	71	#print("%s: deleted" % self.__class__.__name__)
	72	try:
	73	self.conn.close()
	74	self.conn = None
	75	except AttributeError:
	76	pass
	77	except:
	78	raise