def before_request(): """Make sure we are connected to the database each request.""" db.init(app.config['MONGODB_HOST'], app.config['MONGODB_PORT'], 'conversion') try: session['logged_in'] except(KeyError): session['logged_in'] = False
def before_request(): """Make sure we are connected to the database each request.""" db.init(app.config['MONGODB_HOST'], app.config['MONGODB_PORT'], 'conversion') try: session['logged_in'] except (KeyError): session['logged_in'] = False
def transfer(): user_lookup = {} item_lookup = {} rel_lookup = {} global db db = connect_db() mon.init('localhost', 27017, 'conversion') dd = mon.createInitialDb() user_lookup[0] = mon.getUser(dd['admin']) lite_items = query_db('SELECT * FROM items') lite_rels = query_db('SELECT * FROM relations') lite_comments = query_db('SELECT * FROM comments') lite_users = query_db('SELECT * FROM users') lite_votes = query_db('SELECT * FROM votes') for lu in lite_users: print lu['username'] if lu['username'] == 'NKessel': user_lookup[3] = user_lookup[1] # print lu['date_registered'] dr = datetime.strptime(lu['date_registered'], '%Y-%m-%d %H:%M:%S') # print str(dr) u, error = mon.addUser(unicode(lu['username']), unicode(lu['email']), unicode(lu['password']), dr) print 'inserted users: ' + str(u) if u: user_lookup[lu['id']] = u for li in lite_items: # print 'uid lookup: ' + str(li['user_id'])+ ' -- ' + str(user_lookup[li['user_id']]) #print li['time_submitted'] ts = datetime.strptime(li['time_submitted'].split('.')[0], '%Y-%m-%d %H:%M:%S') if not li['body']: li['body'] = li['title'] li['title'] = "" i = mon.addItem({ 'body': unicode(li['body']), 'tldr': unicode(li['title']), 'user': user_lookup[li['user_id']].name, "time_submitted": ts, "upvotes": li['upvotes'], "downvotes": li['downvotes'] }) print 'inserted items: ' + str(li['id']) + ' -- ' + str(i) item_lookup[li['id']] = i for lr in lite_rels: print lr.get('time_submitted', "no time") #tl = datetime.strptime(lr['time_linked'].split('.')[0], '%Y-%m-%d %H:%M:%S') if lr['parent'] == 0: rp = mon.root._id else: rp = item_lookup[lr['parent']]._id if lr['child'] == 0: rc = mon.root._id else: rc = item_lookup[lr['child']]._id lb = user_lookup.get(lr['linked_by'], None) if lb: lb = lb.name else: lb = mon.getUser(item_lookup[lr['child']].user).name r = mon.addRel(rp, rc, lb) r.upvotes = lr['upvotes'] r.downvotes = lr['downvotes'] # r.time_linked = tl r.save() print 'inserted rels: ' + str(lr['id']) + ' -- ' + str(r) rel_lookup[lr['id']] = r for lc in lite_comments: rel = rel_lookup.get(lc['rel_id'], None) if rel is None: continue rel = mon.getRel(rel._id) if lc['head_item'] == 0: cp = mon.root._id else: cp = item_lookup[lc['head_item']]._id rel.comment_parent = cp rel.save() item = mon.getItem(rel.child) item.tags.append(u'comment') item.save() for lv in lite_votes: user = mon.getUser(user_lookup[lv['user_id']].name) rel = rel_lookup.get(lv['rel_id'], None) if rel is None: continue rel_id = rel._id if lv['vote_type'] == 'up': is_upvote = True else: is_upvote = False user.votes.append({ 'rel': rel_id, 'is_upvote': is_upvote }) user.save() db.close()
def transfer(): user_lookup = {} item_lookup = {} rel_lookup = {} global db db = connect_db() mon.init("localhost", 27017, "conversion") dd = mon.createInitialDb() user_lookup[0] = mon.getUser(dd["admin"]) lite_items = query_db("SELECT * FROM items") lite_rels = query_db("SELECT * FROM relations") lite_comments = query_db("SELECT * FROM comments") lite_users = query_db("SELECT * FROM users") lite_votes = query_db("SELECT * FROM votes") for lu in lite_users: print lu["username"] if lu["username"] == "NKessel": user_lookup[3] = user_lookup[1] # print lu['date_registered'] dr = datetime.strptime(lu["date_registered"], "%Y-%m-%d %H:%M:%S") # print str(dr) u, error = mon.addUser(unicode(lu["username"]), unicode(lu["email"]), unicode(lu["password"]), dr) print "inserted users: " + str(u) if u: user_lookup[lu["id"]] = u for li in lite_items: # print 'uid lookup: ' + str(li['user_id'])+ ' -- ' + str(user_lookup[li['user_id']]) # print li['time_submitted'] ts = datetime.strptime(li["time_submitted"].split(".")[0], "%Y-%m-%d %H:%M:%S") if not li["body"]: li["body"] = li["title"] li["title"] = "" i = mon.addItem( { "body": unicode(li["body"]), "tldr": unicode(li["title"]), "user": user_lookup[li["user_id"]].name, "time_submitted": ts, "upvotes": li["upvotes"], "downvotes": li["downvotes"], } ) print "inserted items: " + str(li["id"]) + " -- " + str(i) item_lookup[li["id"]] = i for lr in lite_rels: print lr.get("time_submitted", "no time") # tl = datetime.strptime(lr['time_linked'].split('.')[0], '%Y-%m-%d %H:%M:%S') if lr["parent"] == 0: rp = mon.root._id else: rp = item_lookup[lr["parent"]]._id if lr["child"] == 0: rc = mon.root._id else: rc = item_lookup[lr["child"]]._id lb = user_lookup.get(lr["linked_by"], None) if lb: lb = lb.name else: lb = mon.getUser(item_lookup[lr["child"]].user).name r = mon.addRel(rp, rc, lb) r.upvotes = lr["upvotes"] r.downvotes = lr["downvotes"] # r.time_linked = tl r.save() print "inserted rels: " + str(lr["id"]) + " -- " + str(r) rel_lookup[lr["id"]] = r for lc in lite_comments: rel = rel_lookup.get(lc["rel_id"], None) if rel is None: continue rel = mon.getRel(rel._id) if lc["head_item"] == 0: cp = mon.root._id else: cp = item_lookup[lc["head_item"]]._id rel.comment_parent = cp rel.save() item = mon.getItem(rel.child) item.tags.append(u"comment") item.save() for lv in lite_votes: user = mon.getUser(user_lookup[lv["user_id"]].name) rel = rel_lookup.get(lv["rel_id"], None) if rel is None: continue rel_id = rel._id if lv["vote_type"] == "up": is_upvote = True else: is_upvote = False user.votes.append({"rel": rel_id, "is_upvote": is_upvote}) user.save() db.close()