forked from orionmelt/sherlock
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sub_data.py
37 lines (32 loc) · 1.18 KB
/
sub_data.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
# -*- coding: utf-8 -*-
import csv
subreddits = []
"""
CSV file has the following columns:
name - Subreddit name
topic_level1 - Level 1 topic. For instance, Entertainment.
topic_level2 - Level 2 topic. For instance, TV Shows.
topic_level3 - Level 2 topic. For instance, Sherlock.
default - Y if default sub, blank otherwise.
ignore_text - Y if text in sub needs to be ignored, blank otherwise.
sub_attribute - An attribute we can derive from this subreddit.
For instance, gender, religion, gadget, etc.
sub_value - Value for the above attribute.
For instance, male, atheism, iPhone, etc.
"""
subreddits_file = open("subreddits.csv","r")
for (
name, topic_level1, topic_level2, topic_level3,
default, ignore_text, sub_attribute, sub_value
) in csv.reader(subreddits_file, delimiter=',', quoting=csv.QUOTE_NONE):
subreddit = {
"name" : name,
"topic_level1" : topic_level1,
"topic_level2" : topic_level2,
"topic_level3" : topic_level3,
"default" : default,
"ignore_text" : ignore_text,
"attribute" : sub_attribute.lower(),
"value" : sub_value.lower()
}
subreddits.append(subreddit)