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
