-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathJSONBuilder.py
46 lines (40 loc) · 1.1 KB
/
JSONBuilder.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import sqlite3
import json
class Object:
def toJSON(self):
return json.dumps(self, default=lambda o: o.__dict__,
sort_keys=True, indent=4)
db = sqlite3.connect('tracker.db')
cur = db.cursor()
cur.execute("SELECT * FROM users")
rows = cur.fetchall()
with open("out.txt","w") as file:
tagged = []
for row in rows:
if not row[0] in tagged:
tagged.append(row[0])
color = "red"
if row[1] == "the_donald":
color = "orange"
file.writelines(',"tag.'+row[0]+'":{"color":"' + color + '","link":"'+row[2]+'","tag":"/r/'+row[1]+' user"}')
print("Done")
all_subreddits = Object()
cur.execute("SELECT distinct(subreddit) FROM users")
rows = cur.fetchall()
sub_list = []
for row in rows:
cur.execute('SELECT username,link FROM users where subreddit=?',row)
users = cur.fetchall()
sub = Object()
sub.Name = row[0]
sub_users = []
for user in users:
u = Object()
u.name = user[0]
u.link = user[1]
sub_users.append(u)
sub.Users = sub_users
sub_list.append(sub)
all_subreddits.subreddits = sub_list
with open("subs.json","w") as file:
file.writelines(all_subreddits.toJSON())