def getTestByIdTest(self, id): con = None try: con = psycopg2.connect( user=self.conn["user"], password=self.conn["password"], host=self.conn["host"], port=self.conn["port"], database=self.conn["database"], ) cur = con.cursor() sql = "select * from test where idOfTest = %s" cur.execute(sql, (id, )) con.commit() row = cur.fetchone() if row: r = tes() r.parseTest(row) con.close() return r.toJson() con.close() return "Fail" except (Exception, psycopg2.DatabaseError) as error: return str(error) finally: if con is not None: con.close()
def getTestById(self, id): con = None try: con = psycopg2.connect( user=self.conn["user"], password=self.conn["password"], host=self.conn["host"], port=self.conn["port"], database=self.conn["database"], ) cur = con.cursor() sql = "select * from test where idOfUser = %s" cur.execute(sql, (id, )) con.commit() rows = cur.fetchall() ans = [] for row in rows: r = tes() r.parseTest(row) ans.append(r.toJson()) con.close() return ans except (Exception, psycopg2.DatabaseError) as error: return str(error) finally: if con is not None: con.close()
def get_lineups(): lineups = [] while len(lineups) < 100: rb1 = random.choice(test.rb1s()) print rb1 rb2 = random.choice(test.rb1s()) print rb2 wr1 = random.choice(test.wr1s()) print wr1 wr2 = random.choice(test.wr1s()) print wr2 wr3 = random.choice(test.wr1s()) print wr3 te = random.choice(test.tes()) print te for x in test.qbs(): results = connect("SELECT qb,rb1,rb2,wr1,wr2,wr3,te,flex,dst,oppscore FROM teams WHERE qb=? and wr1=? and wr2=? and wr3=? and rb1=? and rb2=? and te=? ORDER BY oppscore DESC LIMIT 1", (x,wr1,wr2,wr3,rb1,rb2,te)) if results: for row in results: lineups.append(row) print len(lineups) return sorted(lineups, key=lambda k: k[9], reverse=True)
def insert(self, data): con = None try: con = psycopg2.connect( user=self.conn["user"], password=self.conn["password"], host=self.conn["host"], port=self.conn["port"], database=self.conn["database"], ) cur = con.cursor() sql = "INSERT INTO test (timeStart, timeFinish, status, nameTest, numOfQuestion, isEnable, idOfUser, passwdOfTest,limitOfNumUser) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s) " result = ( data.timeStart, data.timeFinish, data.status, data.nameTest, data.numOfQuestion, data.isEnable, data.idOfUser, data.passwdOfTest, data.limitOfNumUser, ) cur.execute(sql, result) con.commit() sql1 = "SELECT * FROM test ORDER BY idOfTest DESC LIMIT 1" cur.execute(sql1) con.commit() row = cur.fetchone() if row: c = tes() c.parseTest(row) con.close() return c.toJson() con.close() return "Fail" except (Exception, psycopg2.DatabaseError) as error: return str(error) finally: if con is not None: con.close()
import sqlite3 as lite import json,csv,test,operator,random allplayers = test.qbs() + test.rb1s() + test.rb2s() + test.wr1s() + test.wr2s() + test.wr3s() + test.tes() def connect(command,options): con = lite.connect('dksalary.db') with con: #con.row_factory = lite.row cur = con.cursor() cur.execute(command,options) rows = cur.fetchall() return rows def WriteListToCSV(csv_file,csv_columns,data_list): try: with open(csv_file, 'w') as csvfile: writer = csv.writer(csvfile, dialect='excel', quoting=csv.QUOTE_NONNUMERIC) writer.writerow(csv_columns) for data in data_list: writer.writerow(data) except IOError as (errno, strerror): print("I/O error({0}): {1}".format(errno, strerror)) return def get_lineups(): lineups = [] while len(lineups) < 100:
import test print(__name__) print(test.tes())