__init__.py 715 B

123456789101112131415161718192021222324
  1. import oursql
  2. from config import db as dbconfig
  3. conn = oursql.connect(host=dbconfig.host, db=dbconfig.database, user=dbconfig.user, passwd=dbconfig.password, autoreconnect=True)
  4. def cursor():
  5. return conn.cursor(oursql.DictCursor)
  6. def execute(cursor, sql, *values):
  7. cursor.execute(sql, values)
  8. def query(cursor, sql, *values):
  9. execute(cursor, sql, *values)
  10. return cursor.fetchall()
  11. def get(cursor, sql, *values):
  12. execute(cursor, sql, *values)
  13. result = cursor.fetchone()
  14. if result is None:
  15. raise Exception('no rows returned for query: {} with values {}'.format(sql, values))
  16. if cursor.fetchone() is not None:
  17. raise Exception('multiple results returned for query'.format(sql, values))
  18. return result