def convert_to_sqlite(): """ :type session : sqlalchemy.orm.session :return: """ import cPickle as Pickle from storage import BVideo, get_session session = get_session('sqlite:///newbilibili.sqlite') data = Pickle.load(open('fraw.data', 'rb')) count = 1 total = len(data) for video in data: #print video[0] b = session.query(BVideo).filter_by(av=video[0]).first() b.tags = video[1] b.description = video[2] b.type = video[3] session.flush() session.commit() if count % 100 == 0: print "[info] %(count)s/%(total)s inserted!" % { 'count': count, 'total': total, } count += 1
def convert_to_sqlite(): """ :type session : sqlalchemy.orm.session :return: """ import cPickle as Pickle from storage import BVideo, get_session session = get_session('sqlite:///newbilibili.sqlite') data = Pickle.load(open('fraw.data', 'rb')) count = 1 total = len(data) for video in data: #print video[0] b = session.query(BVideo).filter_by(av=video[0]).first() b.tags = video[1] b.description = video[2] b.type = video[3] session.flush() session.commit() if count % 100 == 0: print "[info] %(count)s/%(total)s inserted!" % {'count': count, 'total': total, } count += 1
def verify_login(request): key = request.cookies.get("session_key") if not key or not storage.get_session(key): response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) response.set_cookie("message", "User is not logged in.") return response return None
def test_sessions(): test_session = "session." + str(time.time()) s = storage.get_session(test_session) assert s == None storage.add_session({'key': test_session}) s = storage.get_session(test_session) assert type(s) is dict assert 'key' in s assert s['key'] == test_session storage.update_session(test_session, {'elephant': 12}) s = storage.get_session(test_session) assert type(s) is dict assert 'key' in s assert s['key'] == test_session assert 'elephant' in s assert s['elephant'] == 12 storage.delete_session(test_session) s = storage.get_session(test_session) assert s == None
def post_notes(): response = verify_login(request) if response: return response key = request.cookies.get("session_key") session = storage.get_session(key) note = request.form.get("note") if note != None and note != "": storage.add_note({'user': str(session['user']), 'text': str(note)}) response = make_response(redirect("/notes")) response.set_cookie("session_key", key, max_age=600) response.set_cookie("message", "", expires=0) return response
def get_notes(): response = verify_login(request) if response: return response message = request.cookies.get("message") key = request.cookies.get("session_key") session = storage.get_session(key) response = make_response( render_template("notes.html", message=message, session=session)) storage.update_session(key, {"pages": (session.get("pages", 0) + 1)}) response.set_cookie("session_key", key, max_age=600) response.set_cookie("message", "", expires=0) return response
def post_profile(): response = verify_login(request) if response: return response message = request.cookies.get("message") key = request.cookies.get("session_key") session = storage.get_session(key) user = session.get("user") email = request.form.get("email") zipcode = request.form.get("zipcode") storage.update_profile(user, {"email": email, "zipcode": zipcode}) response = make_response(redirect("/notes")) response.set_cookie("session_key", key, max_age=600) response.set_cookie("message", "", expires=0) return response
def get_profile(): response = verify_login(request) if response: return response message = request.cookies.get("message") key = request.cookies.get("session_key") session = storage.get_session(key) user = session.get("user") profile = storage.get_profile(user) response = make_response( render_template("profile.html", message=message, session=session, profile=profile)) response.set_cookie("session_key", key, max_age=600) response.set_cookie("message", "", expires=0) return response
def get_logout(): key = request.cookies.get("session_key") # create a rejection response response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) if not key: response.set_cookie("message", "User is not logged in.") return response session = storage.get_session(key) if not session: response.set_cookie("message", "User is not logged in.") return response if key: storage.delete_session(key) response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) response.set_cookie("message", "", expires=0) return response
def get_notes(): message = request.cookies.get("message") key = request.cookies.get("session_key") # create a rejection response response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) if not key: response.set_cookie("message", "User is not logged in.") return response session = storage.get_session(key) if not session: response.set_cookie("message", "User is not logged in.") return response response = make_response( render_template("notes.html", message=message, session=session)) storage.update_session(key, {"pages": (session.get("pages", 0) + 1)}) response.set_cookie("session_key", key, max_age=600) response.set_cookie("message", "", expires=0) return response
def post_notes(): key = request.cookies.get("session_key") # create a rejection response response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) if not key: response.set_cookie("message", "User is not logged in.") return response session = storage.get_session(key) if not session: response.set_cookie("message", "User is not logged in.") return response # OK, we are logged in so process the form submission user = request.form.get("user") email = request.form.get("email") zip = request.form.get("zip") note = request.form.get("note") if note != None and note != "": storage.add_note({'text': str(user + ": " + note)}) response = make_response(redirect("/notes")) response.set_cookie("session_key", key, max_age=600) response.set_cookie("message", "", expires=0) return response
def post_udashboard(): #get session key from users cookie. Could be dangerous. BE CAREFUL!! key = request.cookies.get("session_key") #response used if checks fail response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) #check for key and session if not key: return response session = storage.get_session(key) if not session: return response #TODO: Something to look into: Might need to determine which button was clicked first #schedule/edit a call response = make_response( render_template("user_dashboard.html", session=session)) storage.update_session(key) response.set_cookie("session_key", key, max_age=600) return response
def get_udashboard(): key = request.cookies.get("session_key") #response used if checks fail response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) #check for key and session if not key: return response session = storage.get_session(key) if not session: return response #variable to display scheduled calls email = session['email'] calls = storage.customer_scheduled_calls(email) #displaying page and updating session response = make_response( render_template("user_dashboard.html", session=session), calls) storage.update_session(key) response.set_cookie("session_key", key, max_age=600) return response
def get_pdashboard(): #get session key from users cookie. Could be dangerous. BE CAREFUL!! key = request.cookies.get("session_key") #response used if checks fail response = make_response(redirect("/login")) response.set_cookie("session_key", "", expires=0) #check for key and session if not key: return response session = storage.get_session(key) if not session: return response #view scheduled calls email = session['email'] calls = storage.provider_scheduled_calls(email) response = make_response( render_template("provider_dashboard.html", session=session), calls) storage.update_session(key) response.set_cookie("session_key", key, max_age=600) return response