summaryrefslogtreecommitdiff
path: root/populate_category.py
diff options
context:
space:
mode:
Diffstat (limited to 'populate_category.py')
-rw-r--r--populate_category.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/populate_category.py b/populate_category.py
new file mode 100644
index 0000000..339811c
--- /dev/null
+++ b/populate_category.py
@@ -0,0 +1,26 @@
+import os
+import datetime
+from datetime import timedelta
+
+Date = datetime.datetime.now()
+def populate():
+ #file_obj=open("data_of_rc.csv")
+ #print file_obj
+ #file_obj.close()
+ f=open("category_names.txt")
+ for line in f:
+ cname=line.replace("\n","")
+ cat_added=add_category(cname)
+ f.close()
+
+def add_category(cname):
+ cat = FossCategory.objects.get_or_create(name=cname,description=cname,date_created=Date,date_modified=Date)[0]#for now description is same as category_names
+ print "category " +str(cname)+" added at "+str(Date)
+ return cat
+
+# Start execution here!
+if __name__ == '__main__':
+ print "Starting population script for adding category..."
+ os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'forums.settings')
+ from website.models import *
+ populate()