simon987/od-database
1
0
Fork 0
mirror of https://github.com/simon987/od-database.git synced 2025-12-13 23:09:01 +00:00
Code Issues Projects Releases Wiki Activity

Bug fixes

Browse Source
This commit is contained in:
simon987
2019-03-28 20:29:34 -04:00
parent d69ed65a0c
commit 2046b36f9a
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
database.py
View File

@@ -84,7 +84,7 @@ class Database:
(url,))
db_web = cursor.fetchone()
if db_web:
website = Website(db_web[1], db_web[2], db_web[3], db_web[4], int(db_web[0].timestamp()))
website = Website(db_web[1], db_web[2], db_web[3], db_web[4], str(db_web[0]))
return website
else:
return None
@@ -98,7 +98,7 @@ class Database:
db_web = cursor.fetchone()
if db_web:
website = Website(db_web[1], db_web[2], db_web[3], int(db_web[4].timestamp()))
website = Website(db_web[1], db_web[2], db_web[3], str(db_web[4]))
website.id = db_web[0]
return website
else:
@@ -119,8 +119,7 @@ class Database:
with psycopg2.connect(self.db_conn_str) as conn:
cursor = conn.cursor()
cursor.execute(
"SELECT id FROM Website WHERE id >= (abs(random()) % (SELECT max(id) FROM Website)) LIMIT 1;")
cursor.execute("SELECT id FROM Website ORDER BY random() LIMIT 1")
return cursor.fetchone()[0]
@@ -325,4 +324,3 @@ class Database:
logged_useragent=None
)
for r in cursor.fetchall()]