def add_users():
user_id = fk.request.cookies.get('user_id')
if fk.request.method == 'POST':
print (fk.request.form)
if "update" in fk.request.form:
print ('update in')
user_data = {}
user_data["firstname"] = fk.request.form["firstname"]
user_data["lastname"] = fk.request.form["lastname"]
user_data["email"] = fk.request.form["email"]
if 'nonharvard' in fk.request.form:
user_data['nonharvard'] = True
else:
user_data['nonharvard'] = False
if 'choreographer' in fk.request.form:
user_data['choreographer'] = True
else:
user_data['choreographer'] = False
dm.add_user(user_data)
db_list = dm.get_user_data_list()
return fk.render_template('add_user.html', data_list = db_list, user_name=user_id)
elif fk.request.method == 'GET':
print ('fk get')
db_list = dm.get_user_data_list()
return fk.render_template('add_user.html', data_list = db_list, user_name=user_id)
def create_new_class(self, create_class_list_widget, class_name):
# TO DO: creating class and then immediately removing class doesn't work
if not len(create_class_list_widget.student_widgets):
print('Class is empty!')
return
path = os.path.join('records',
class_name.strip().replace(' ', '_') + '.db')
if os.path.exists(path):
print('Error! Class already exists!')
return
print('Creating New Class:', path)
db = open(path, 'w')
conn = sqlite3.connect(path)
curs = conn.cursor()
db_cons.append(conn)
db_curs.append(curs)
db_manager.reset(curs, True)
for i, w in enumerate(create_class_list_widget.student_widgets):
print(w.name)
db_manager.add_user(curs, i, w.name)
tab = TabbedPanelHeader(text=class_name)
tab.content = StudentList(len(self.student_main.class_tabs))
self.student_main.class_tabs.append(tab)
self.student_main.add_widget(tab)
# db_manager.close(conn)
self.dismiss_create_class_popup()
def start_callback(bot, update):
if (update.message.chat_id < 0):
bot.send_message(chat_id=update.message.chat_id,
text="Hello! Click start to allow me to pm you",
reply_markup=reply_markup)
else:
username = update.message.from_user.username
print(username)
print(username is None)
if username is None:
bot.send_message(
chat_id=update.message.chat_id,
text=
"You need to set a username to use this bot. Set and start again."
)
else:
greeting = "Hello {}, you've been successfully added to u-l8-m8-bot!".format(
username)
add_user(update.message.from_user.username, update.message.chat_id)
bot.send_message(chat_id=update.message.chat_id, text=greeting)
def GET(self, request_str):
req_valid, req_type, req_arg = self.parse_request(request_str)
if req_valid == False:
return 'Invalid request'
else:
# REQUEST NODE INFO
if req_type == 'RESET':
db_manager.remove_server()
db_manager.remove_server_for_cache()
db_manager.remove_all_videos()
db_manager.remove_all_nodes()
elif req_type == 'GET_SERVER_ADDRESS':
res = db_manager.get_server()
return str(res[0].ip) + ' ' + str(res[0].port)
elif req_type == 'GET_SERVER_ADDRESS_FOR_CACHE':
print 'get_server_address_for_cache'
res = db_manager.get_server_for_cache()
return str(res[0].ip) + ' ' + str(res[0].port)
elif req_type == 'GET_CACHES_ADDRESS':
# req = "user-hyunah-1 & 10"
arg_user_name = req_arg.split('_')[0]
arg_num_of_caches = req_arg.split('_')[1]
n_of_current_caches = db_manager.get_num_of_caches()
n_of_returned_caches = min(n_of_current_caches, int(arg_num_of_caches))
print '[tracker.py] n_of_returned_caches', n_of_returned_caches
caches = db_manager.get_many_caches(arg_user_name, n_of_returned_caches)
ret_str = ''
for cache in caches:
ret_str = ret_str + str(cache.ip) + ' ' + str(cache.port) + '\n'
return ret_str
# NODE REGISTER
elif req_type == 'REGISTER_USER':
# req_arg = "143.243.23.13_324"
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
arg_watching_video = req_arg.split('_')[2]
db_manager.add_user(arg_ip, arg_port, arg_watching_video)
print '[tracker.py] Accessing...'
print '[tracker.py] user_pop', user_population
user_population[str(arg_watching_video)] += 1
log_load()
return 'User is registered'
elif req_type == 'REGISTER_CACHE':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
db_manager.add_cache(arg_ip, arg_port)
return 'Cache is registered'
elif req_type == 'REGISTER_SERVER':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
# remove existing server & videos
db_manager.add_server(arg_ip, arg_port)
return 'Server is registered'
elif req_type == 'REGISTER_SERVER_FOR_CACHE':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
# remove existing server & videos
db_manager.add_server_for_cache(arg_ip, arg_port)
return 'Server is registered'
# VIDEO REGISTER
elif req_type == 'REGISTER_VIDEO':
print 'add video'
split_arg = req_arg.split('_')
arg_vname = split_arg[0]
arg_n_of_frames = split_arg[1]
arg_code_param_n = split_arg[2]
arg_code_param_k = split_arg[3]
arg_total_size = split_arg[4]
arg_chunk_size = split_arg[5]
arg_last_chunk_size = split_arg[6]
db_manager.add_video(arg_vname, arg_n_of_frames, arg_code_param_n, arg_code_param_k, arg_total_size, arg_chunk_size, arg_last_chunk_size)
if str(arg_vname) not in user_population.keys():
user_population[str(arg_vname)] = 0
print '[tracker.py] arg_vname', str(arg_vname)
print '[tracker.py] user_pop', user_population
return 'Video is registered'
elif req_type == 'GET_ALL_VIDEOS':
videos = db_manager.get_all_videos()
ret_str = ''
for video in videos:
ret_str = ret_str + str(video.id) + ' ' + str(video.vname) + ' ' + str(video.n_of_frames) + ' ' + str(video.code_param_n) + ' ' + str(video.code_param_k) + ' ' + str(video.total_size) + ' ' + str(video.chunk_size) + ' ' + str(video.last_chunk_size) + '\n'
return ret_str
elif req_type == 'REMOVE_SERVER':
db_manager.remove_server()
return 'Server is removed'
elif req_type == 'REMOVE_SERVER_FOR_CACHE':
db_manager.remove_server_for_cache()
return 'Server for cache is removed'
elif req_type == 'REMOVE_USER':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
arg_watching_video = req_arg.split('_')[2]
db_manager.remove_user(arg_ip, arg_port, arg_watching_video)
user_population[str(arg_watching_video)] -= 1
log_load()
return 'User is removed'
elif req_type == 'REMOVE_CACHE':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
db_manager.remove_cache(arg_ip, arg_port)
return 'Cache is removed'
elif req_type == 'UPDATE_CHUNKS_FOR_CACHE':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
arg_vname = req_arg.split('_')[2]
arg_chunk_str = req_arg.split('_')[3]
db_manager.add_chunks_for_cache(arg_ip, arg_port, arg_vname, arg_chunk_str)
elif req_type == 'UPDATE_SERVER_LOAD':
arg_vname = req_arg.split('_')[0]
arg_n_of_chks = req_arg.split('_')[1]
db_manager.add_server_load(arg_vname, arg_n_of_chks)
def GET(self, request_str):
req_valid, req_type, req_arg = self.parse_request(request_str)
if req_valid == False:
return 'Invalid request'
else:
# REQUEST NODE INFO
if req_type == 'RESET':
db_manager.remove_server()
db_manager.remove_server_for_cache()
db_manager.remove_all_videos()
db_manager.remove_all_nodes()
db_manager.remove_all_caches_from_account_cache()
elif req_type == 'GET_SERVER_ADDRESS':
if session.get('login', False):
res = db_manager.get_server()
return str(res[0].ip) + ' ' + str(res[0].port)
else:
raise web.seeother('/login')
elif req_type == 'GET_SERVER_ADDRESS_FOR_CACHE':
if session.get('login', False):
print 'get_server_address_for_cache'
res = db_manager.get_server_for_cache()
return str(res[0].ip) + ' ' + str(res[0].port)
else:
raise web.seeother('/login')
elif req_type == 'GET_CACHES_ADDRESS':
if session.get('login', False):
# req = "user-hyunah-1 & 10"
arg_user_name = req_arg.split('_')[0]
arg_num_of_caches = req_arg.split('_')[1]
n_of_current_caches = db_manager.get_num_of_caches()
n_of_returned_caches = min(n_of_current_caches, int(arg_num_of_caches))
print '[tracker.py] n_of_returned_caches', n_of_returned_caches
caches = db_manager.get_many_caches(arg_user_name, n_of_returned_caches)
ret_str = ''
for cache in caches:
ret_str = ret_str + str(cache.ip) + ' ' + str(cache.port) + '\n'
return ret_str
else:
raise web.seeother('/login')
# NODE REGISTER
elif req_type == 'REGISTER_USER':
if session.get('login', False):
un = session.user_name
# req_arg = "143.243.23.13_324"
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
arg_watching_video = req_arg.split('_')[2]
#INSERT HERE
#Get the list of owned videos here
owned_videos= db_manager.get_owned_videos(un)
if arg_watching_video not in owned_videos:
#If requested video is not in the list, subtract 5 points
res = db_manager.update_owned_videos(un, arg_watching_video)
#If not enough points, return that says so
if res == 'Not enough points':
return 'Not enough points'
db_manager.add_user(arg_ip, arg_port, arg_watching_video)
print '[tracker.py] Accessing...'
print '[tracker.py] user_pop', user_population
user_population[str(arg_watching_video)] += 1
log_load()
return 'User is registered'
else:
raise web.seeother('/login')
elif req_type == 'REGISTER_CACHE':
if session.get('login', False):
##
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
multiplier = req_arg.split('_')[2]
#arg_config = req_arg.split('_')[2] TODO: cache size
db_manager.add_cache(arg_ip, arg_port)
un = session.user_name
db_manager.add_cache_to_account_cache(un, arg_ip, arg_port, multiplier)
return 'Cache is registered'
else:
raise web.seeother('/login')
elif req_type == 'REGISTER_SERVER':
#CURRENTLY NO SERVER ACCOUNT BUT WE SHOULD HAVE ONE
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
# remove existing server & videos
db_manager.add_server(arg_ip, arg_port)
return 'Server is registered'
elif req_type == 'REGISTER_SERVER_FOR_CACHE':
#CURRENTLY NO SERVER ACCOUNT BUT WE SHOULD HAVE ONE
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
# remove existing server & videos
db_manager.add_server_for_cache(arg_ip, arg_port)
return 'Server is registered'
# VIDEO REGISTER
elif req_type == 'REGISTER_VIDEO':
#CURRENTLY NO SERVER ACCOUNT BUT WE SHOULD HAVE ONE
print 'add video'
split_arg = req_arg.split('_')
arg_vname = split_arg[0]
arg_n_of_frames = split_arg[1]
arg_code_param_n = split_arg[2]
arg_code_param_k = split_arg[3]
arg_total_size = split_arg[4]
arg_chunk_size = split_arg[5]
arg_last_chunk_size = split_arg[6]
db_manager.add_video(arg_vname, arg_n_of_frames, arg_code_param_n, arg_code_param_k, arg_total_size, arg_chunk_size, arg_last_chunk_size)
if str(arg_vname) not in user_population.keys():
user_population[str(arg_vname)] = 0
print '[tracker.py] arg_vname', str(arg_vname)
print '[tracker.py] user_pop', user_population
return 'Video is registered'
elif req_type == 'GET_OWNED_VIDEOS':
if session.get('login', False):
un = session.user_name
owned_videos= db_manager.get_owned_videos(un)
##
web.header('Content-Type', 'application/json')
return json.dumps(owned_videos)
else:
raise web.seeother('/login')
elif req_type == 'GET_ALL_VIDEOS':
#users can still get all videos
videos = db_manager.get_all_videos()
ret_str = ''
for video in videos:
ret_str = ret_str + str(video.id) + ' ' + str(video.vname) + ' ' + str(video.n_of_frames) + ' ' + str(video.code_param_n) + ' ' + str(video.code_param_k) + ' ' + str(video.total_size) + ' ' + str(video.chunk_size) + ' ' + str(video.last_chunk_size) + '\n'
return ret_str
elif req_type == 'REMOVE_SERVER':
db_manager.remove_server()
return 'Server is removed'
elif req_type == 'REMOVE_SERVER_FOR_CACHE':
db_manager.remove_server_for_cache()
return 'Server for cache is removed'
elif req_type == 'REMOVE_USER':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
arg_watching_video = req_arg.split('_')[2]
db_manager.remove_user(arg_ip, arg_port, arg_watching_video)
user_population[str(arg_watching_video)] -= 1
log_load()
return 'User is removed'
elif req_type == 'REMOVE_CACHE':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
db_manager.remove_cache(arg_ip, arg_port)
##
un = session.user_name
#Right now not using username as input, because the server can remove any user it wants.
#Need to have server logged in as admin for its extra priverlages
db_manager.remove_cache_from_account_cache(ip=arg_ip, port=arg_port)
return 'Cache is removed'
elif req_type == 'UPDATE_CHUNKS_FOR_CACHE':
arg_ip = req_arg.split('_')[0]
arg_port = req_arg.split('_')[1]
arg_vname = req_arg.split('_')[2]
arg_chunk_str = req_arg.split('_')[3]
db_manager.add_chunks_for_cache(arg_ip, arg_port, arg_vname, arg_chunk_str)
elif req_type == 'UPDATE_SERVER_LOAD':
arg_vname = req_arg.split('_')[0]
arg_n_of_chks = req_arg.split('_')[1]
db_manager.add_server_load(arg_vname, arg_n_of_chks)
elif req_type == 'CACHE_DATA_VIS':
return render.user_to_cache_data()
elif req_type == 'CACHE_TO_USER_DATA':
return render.cache_to_user_data()
elif req_type =='GET_CACHE_DATA':
web.header('Content-Type', 'application/json')
user_data = dv.get_user_logs_as_json()
return json.dumps(user_data)
#currently reading from file. must later have post request to store data into db
elif req_type =='GET_CACHE_DATA2':
web.header('Content-Type', 'application/json')
user_data = dv.get_user_logs_as_json()
user_data = dv.rearrange_data_for_caches(user_data)
return json.dumps(user_data)
#currently reading from file. must later have post request to store data into db
elif req_type == 'VERIFY_ACCOUNT':
username = req_arg.split('_')[0]
password = req_arg.split('_')[1]
if len(db_manager.verify_account(username, password)) == 0:
return False
else:
return True