def issue_norm_stance():
for issue in db.ISSUE.find({}):
if isinstance(issue["norm"], list):
norm_dict = dict(zip(["side", "importance", "issue", "source_db"], issue["norm"] + ["issue"]))
norm_stance = Stance(**norm_dict)
encoded_stance = db.encode_classes(norm_stance)
db.ISSUE.update(issue, {"$set": {"norm": encoded_stance}})
print "converted norm stances to stance objects"
from pymongo import MongoClient
from bson.json_util import loads
import pol.database as db
from pol.stance import Stance
for COLLECTION in db.collection_list:
result = []
for data in open('mongo_dumps/' + COLLECTION.name + ".txt").read().splitlines():
result.append(loads(data))
for hash_data in result:
if COLLECTION.name == 'issue':
if "norm" in hash_data and hash_data["norm"]:
hash_data["norm"] = db.encode_classes(Stance(**hash_data["norm"]))
if not COLLECTION.find_one({"_id": hash_data["_id"]}):
COLLECTION.insert(hash_data)
else:
COLLECTION.update({"_id": hash_data["_id"]}, hash_data)
print COLLECTION.name, "loaded!"
