def populate_tables():
print("Populating the Activities table..")
new_activity = Activity(name="Sleep", user_id=0, color="lightgrey")
session.add(new_activity)
new_activity = Activity(name="Work - Meeting", user_id=0, color="#2196f3")
session.add(new_activity)
new_activity = Activity(name="Work - Coding", user_id=0, color="#03a9f4")
session.add(new_activity)
new_activity = Activity(name="Coding", user_id=0, color="lightblue")
session.add(new_activity)
new_activity = Activity(name="Exercise", user_id=0, color="red")
session.add(new_activity)
new_activity = Activity(name="Transport", user_id=0, color="green")
session.add(new_activity)
new_activity = Activity(name="Entertainment", user_id=0, color="peachpuff")
session.add(new_activity)
new_activity = Activity(name="Internet", user_id=0, color="silver")
session.add(new_activity)
new_activity = Activity(name="Downtime", user_id=0, color="grey")
session.add(new_activity)
new_activity = Activity(name="Socialization", user_id=0, color="yellow")
session.add(new_activity)
session.commit()
password, salt = User.hash_password("admin")
new_user = User(
name="Admin",
username="******",
password=password,
salt=salt,
is_authenticated=False,
)
session.add(new_user)
session.commit()
print("Done populating")
def login():
form = LoginForm()
if 'username' in session:
username = session['username']
return 'Logged in as ' + username + '<br>' + "<b><a href = '/logout'>click here to log out</a></b>"
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
login_success = User.authenticate_user(username, password)
if login_success:
# Successful login
session['username'] = username
session['password'] = password
user_obj = User.get_user(username)
session['name'] = user_obj.name
return redirect(url_for('index'))
else:
# Failed login
unsuccessful = 'Please check your credentials'
return render_template('login.html',
title='Sign In',
form=form,
alertmessage=unsuccessful)
if form.validate_on_submit():
flash('Login requested for user {}, remember_me={}'.format(
form.username.data, form.remember_me.data))
return redirect('/index')
return render_template('login.html', title='Sign In', form=form)
def change_email():
"""
页面:修改账户的电子邮寄地址
:return:
"""
if session['email']:
if request.method == 'POST':
new_email = request.form["Input_newemail"]
password = request.form["InputPassword"]
if User.exist_user(new_email):
message = "您输入的新email已经被人使用!"
return render_template("change_email.html", message=message)
result = User.check_user(session['email'], password)
if result is True:
User.update_user_email(session['email'], new_email)
session['email'] = new_email
message = "您的账号邮箱已经更改为{}".format(new_email)
return render_template("change_email.html", message=message)
else:
message = "您的密码输入错误!"
return render_template("change_email.html", message=message)
else:
return render_template("change_email.html")
else:
return redirect("/login")
def log_in():
''' For ajax validation of login information '''
email = request.form.get('username')
user_psw = request.form.get('password')
remember_me = request.form.get('remember')
hashed = User.hash_password(user_psw)
the_user = User.log_in(email, hashed)
if the_user:
if remember_me:
login_user(the_user, remember=True)
else:
login_user(the_user)
flash('Logged in successfully.')
if (the_user.auth_level == 1) or (the_user.auth_level == "1"):
exhibitor = Exhibitor.get_exhibitor(the_user.id)
last_logged_in = exhibitor.get_last_logged_in()
if last_logged_in is None:
exhibitor.set_last_logged_in()
return json.dumps({'logged_in': True, 'url':'/nytt_losenord/'})
exhibitor.set_last_logged_in()
return json.dumps({'logged_in': True, 'url':'/utstallare/'})
return json.dumps({'logged_in': True, 'url':'/administrator/addexhibitor'})
else:
return json.dumps({'logged_in': False})
def main():
get_auth = VKAuth(['friends'], '6892678', '5.95')
get_auth.auth()
print('Получен следующий токен {}'.format(
get_auth._access_token)) # получили токен пользователя
user_input = input('Введите id или имя пользователя: ')
user = User(get_auth._access_token)
user_id = user.check_user(user_input)
user_data = user.get_user_data(user_id)
friends = FriendSearch(get_auth._access_token, user_id)
db = DB_Mongo()
db.all_drop()
potencial_friend_city_sex = get_potencial_friend_city_sex(
friends, user_data) # Получаем список из VK
db.db_operation(db, potencial_friend_city_sex, user_data, friends,
user_id) # запись базовый список в БД
selection_criterion = sort_potencial_friend(
) # формируем уточняющие критерии
out_db = db.find_n_drop_adv(
selection_criterion) # сортируем по уточняющим критериям
out_list = []
for item in out_db.find():
x = friends.get_fr(item['id'])
out_list.append(x)
list_to_json(out_list, 'outjson.json')
pprint(out_list)
def saveSong(songID):
try:
if (songID == "0") or (User.checkForDuplicateSavedSong(session['email'], songID, conn)):
return json.dumps({'savedSong': False})
else:
User.saveSong(session['email'], songID, conn)
return json.dumps({'savedSong': True})
except:
return json.dumps({'savedSong': True})
def load_user(id):
if id is None:
return redirect(url_for('bp_login.login'))
user = User()
user.get_by_id(id)
if user.is_active():
return user
else:
return None
def change_passwd():
params = request.args if request.method == 'GET' else request.form
uid = params.get('userid')
upass = params.get('user-password')
muser = session['username']['username']
mpass = params.get('manager-password')
_is_ok, _error = User.valid_change_passwd(uid, upass, muser, mpass)
if _is_ok:
_is_ok, _error = User.change_passwd(uid, upass)
return jsonify({'is_ok': _is_ok, 'error': _error})
def loginAuth():
email = request.form['email']
password = request.form['password']
if User.validateUser(email, password, conn):
session['email'] = email
session['username'] = User.fetchUsername(email, conn)
return redirect('/home')
else:
error = 'Incorrect Login, Please enter again'
return render_template('login.html', error=error)
def change_passwd():
params = request.args if request.method == 'GET' else request.form
uid = params.get('userid')
upass = params.get('user-password')
muser = session['username']['username']
mpass = params.get('manager-password')
_is_ok,_error = User.valid_change_passwd(uid, upass, muser, mpass)
if _is_ok:
_is_ok,_error = User.change_passwd(uid, upass)
return jsonify({'is_ok':_is_ok,'error':_error})
def test_saveSong():
User.saveSong(email1, songID1, conn)
User.saveSong(email2, songID2, conn)
User.saveSong(email1, songID3, conn)
assert User.getSavedSongs(email1, conn) == [{'email': '*****@*****.**', 'songID': '887fgf7ergf8e'}, {'email': '*****@*****.**', 'songID': 'djhgds7678wev'}]
assert User.getSavedSongs(email2, conn) == [{'email': '*****@*****.**', 'songID': 'chds763ndsuds'}]
assert User.getSavedSongs(email3, conn) == ()
def registerAuth():
email = request.form['email']
username = request.form['username']
password = request.form['password']
if not User.checkIfUserExists(email, conn):
User.insertRegisterDetails(email, username, password, conn)
session['email'] = email
session['username'] = username
return redirect('/home')
else:
error = 'User already exists, enter another email'
return render_template('register.html', error=error)
def done(self):
# make sure the user doesn't already exist
u = User.by_name(self.username)
if u:
msg = 'That user already exists.'
self.render('signup-form.html', error_username = msg)
else:
u = User.register(self.username, self.password, self.email)
u.put()
self.login(u)
self.redirect('/')
def login_method():
if request.method == 'POST':
account = request.form['InputAccount']
password = request.form['InputPassword']
check = User.is_login_valid(account, password)
if check is True:
session['account'] = account
session['name'] = User.find_user_data(account).get('Name')
return redirect("/")
else:
message = "Ur account or password is wrong !!"
return render_template('login.html', message=message)
else:
return render_template('login.html')
def login():
if request.method == 'POST':
email = request.form['InputEmail']
password = request.form['InputPassword']
result = User.check_user(email, password)
if result is True:
session['email'] = email
session['name'] = User.find_user_data(email)['name']
return redirect("/")
else:
message = "Input email or password is incorrect"
return render_template("login.html", message=message)
else:
return render_template("login.html")
def login():
if request.method == 'POST':
email = request.form['InputEmail']
password = request.form['InputPassword']
result = User.check_user(email, password)
if result is True:
session['email'] = email
session['name'] = User.find_user_data(email)['name']
return redirect("/")
else:
message = "Input email or password is incorrect"
return render_template("login.html", message=message)
else:
return render_template("login.html")
def get_user(usr_in):
try:
int(usr_in)
user = User(usr_in)
return user
except ValueError:
params = {'user_ids': usr_in, 'access_token': TOKEN, 'v': 5.92}
try:
response_get_id = requests.get(
'https://api.vk.com/method/users.get', params)
user = User(int(response_get_id.json()['response'][0]
['id'])) # получили по имени пользователя его id
return user
except Exception as e:
print(response_get_id.json()['error']['error_msg'])
def register_method():
if request.method == 'POST':
name = request.form['InputName']
account = request.form['InputAccount']
password = request.form['InputPassword']
result = User.register_user(name, account, password)
if result is True:
session['account'] = account
session['name'] = User.find_user_data(account).get('Name')
return redirect("/")
else:
message = "Ur account is already existed !!"
return render_template('register.html', message=message)
else:
return render_template('register.html')
def logged_user(app):
# user = User(username='******', password='******', real_name='Demo')
user = User(username='******', password='******', real_name='Admin')
# app.login_as(username='******', password='******')
app.login_as(user)
yield user
app.logout()
def main():
""" Entry point of the script. """
client = MongoClient(Constants.MONGO_URI)
db = client[Constants.DATABASE]
# Collections
collection_exercises = db[Constants.COLLECTION_EXERCISE]
collection_folders = db[Constants.COLLECTION_FOLDER]
# Delete user contents entirely before seeding data.
num_deleted = User.delete_all_user_content(collection_folders, collection_exercises)
print "Deleted " + str(num_deleted[0]) + " exercises " + str(num_deleted[1]) + " folders."
created_exercises_ids = Exercise.create_exercises_in_folder(collection_exercises,
collection_folders,
Constants.ROOT_EXERCISES,
Constants.ROOT_FOLDER_ID)
print "Created Root Exercises: %s" % created_exercises_ids
# created_folders_ids = create_folders(folders, root_stack)
# print "Created Root Folders: %s" % created_folders_ids
create_folders_and_exercises(collection_folders,
collection_exercises,
Constants.FOLDER_EXERCISES,
Constants.ROOT_STACK_ID)
create_practice_sessions(collection_folders, collection_exercises)
def index():
if 'username' in session:
user = session['name']
user_obj = User.get_user(session['username'])
else:
user = '******'
return render_template('index.html', user=user, user_obj=user_obj)
def test_login(app):
user = User(username='******', password='******', real_name='Admin')
# user = User(username='******', password='******', real_name='Demo')
# app.login_as(username='******',password ='******')
app.login_as(user)
assert app.is_user_menu_present()
# TODO: add verifficatin of presence User Name
def login_method():
session['inLoginPage'] = True
session['inRegisterPage'] = False
if request.method == 'POST':
account = request.form['InputAccount']
password = request.form['InputPassword']
check = User.is_login_valid(account, password)
if check == True:
session['account'] = account
session['name'] = User.find_user_data(account).get('name')
session['login'] = True
return redirect("/")
else:
message = "Your account or password is invalid !!"
return render_template("login.html", message=message)
else:
return render_template("login.html")
def user_del():
params = request.args if request.method == 'GET' else request.form
id = params.get('id')
username = params.get('username')
if username == 'admin':
return jsonify({'is_ok': False, 'error': '管理员不允许删除'})
_is_ok, _error = User.user_del(int(id), username)
return jsonify({'is_ok': _is_ok, 'error': _error})
def user_del():
params = request.args if request.method == 'GET' else request.form
id = params.get('id')
username = params.get('username')
if username == 'admin':
return jsonify({'is_ok': False, 'error': '管理员不允许删除'})
_is_ok, _error = User.user_del(int(id), username)
return jsonify({'is_ok': _is_ok, 'error': _error})
def register():
data = flask.request.get_json(force=True)
try:
password, salt = User.hash_password(data["password"])
new_user = User(
name=data["name"],
username=data["username"],
password=password,
salt=salt,
is_authenticated=True,
)
db.session.add(new_user)
db.session.commit()
return flask.jsonify(new_user.serialize)
except:
return ("Bad", 400)
def change_email():
if session['email']:
if request.method == 'POST':
new_email = request.form['InputNewEmail']
password = request.form['InputPassword']
result = User.check_user(session['email'], password)
if result:
User.update_user_email(session['email'], new_email)
session['email'] = new_email
message = f"您的邮箱已更新为{new_email}"
return render_template("change_email.html", message=message)
else:
message = "您的密码错误！"
return render_template("change_email.html", message=message)
else:
return render_template("change_email.html")
else:
return redirect('/login')
def change_email():
if session['email']:
if request.method == 'POST':
new_email = request.form['InputNewEmail']
password = request.form['InputPassword']
result = User.check_user(session['email'], password)
if result is True:
User.update_user_email(session['email'], new_email)
session['email'] = new_email
message = "Your new email is {}".format(new_email)
return render_template("change_email.html", message=message)
else:
message = "Input password is incorrect"
return render_template("change_email.html", message=message)
else:
return render_template("change_email.html")
else:
return redirect("/login")
def test_database(self):
database = DatabaseHandler.get_instance()
database.store_user(User(self.test_uid, request_handler.Helper.hash_password(self.test_password)))
assert database.get_user(self.test_uid) is not None, b"Couldn't retrieve user from database!"
user_names = []
for user in database.get_users():
user_names.append(user.uid)
assert self.test_uid in user_names, b"Retrieving all users does not work!"
def change_email():
if session['email']:
if request.method == 'POST':
new_email = request.form['InputNewEmail']
password = request.form['InputPassword']
result = User.check_user(session['email'], password)
if result is True:
User.update_user_email(session['email'], new_email)
session['email'] = new_email
message = "Your new email is {}".format(new_email)
return render_template("change_email.html", message=message)
else:
message = "Input password is incorrect"
return render_template("change_email.html", message=message)
else:
return render_template("change_email.html")
else:
return redirect("/login")
def register_method():
session['inLoginPage'] = False
session['inRegisterPage'] = True
if request.method == 'POST':
name = request.form['InputName']
account = request.form['InputAccount']
password = request.form['InputPassword']
result = User.register_user(name, account, password)
if result == True:
session['account'] = account
session['name'] = User.find_user_data(account).get('name')
return redirect("/")
else:
message = "Your account is already exist !!"
return render_template("register.html", message=message)
else:
return render_template("register.html")
def load_user_from_request(request):
api_key = request.headers.get('Authorization')
if api_key:
obj = jwt_decoding(api_key)
user = obj.get('info', None)
if user:
try:
id = user.get('id', None)
user = User()
user = user.get_by_id(id)
except Exception as e:
logger.error(e)
else:
return user
else:
# print("is exception !!!!" + str(obj['error_msg']))
return None
def get_all_users():
"""
Returns list of all Users
:return: [User]
"""
with lock:
cursor.execute("SELECT * FROM users")
data = cursor.fetchall()
return [User(x) for x in data]
def get_users_with_reminders():
"""
Returns list of Users, who allowed to send them reminders
:return: [User]
"""
with lock:
cursor.execute("SELECT * FROM users WHERE need_reminders=1")
data = cursor.fetchall()
return [User(x) for x in data]
def login():
"""
页面:用户登录
:return:
"""
if request.method == 'POST':
email = request.form["InputEmail"]
password = request.form["InputPassword"]
result = User.check_user(email, password)
if result is True:
# 转跳到home页
session['email'] = email
session['name'] = User.get_user_data(email)['name']
return redirect('/')
else:
message = "您的电子邮箱或者密码错误!"
return render_template("login.html", message=message)
else:
return render_template("login.html")
def login():
params = request.args if request.method == 'GET' else request.form
username = params.get('username', '')
password = params.get('password', '')
# 类调用--------
get_session = User.validate_login(username, password)
if get_session:
session['username'] = get_session
return redirect('/user/')
else:
return render_template('user/login.html', username=username, error='用户认证失败')
def register():
if request.method == 'POST':
name = request.form['InputName']
email = request.form['InputEmail']
password = request.form['InputPassword']
result = User.register_user(name, email, password)
if result is True:
session['email'] = email
session['name'] = name
return redirect("/")
else:
message = "This email is already existed"
return render_template("register.html", message=message)
else:
return render_template("register.html")
def newuser():
params = request.args if request.method == 'GET' else request.form
_is_ok,_error = User.user_add(params)
return jsonify({'is_ok':_is_ok,'error':_error})
def user():
return render_template('user/user.html',user_list=User.get_list())
def user_update():
params = request.args if request.method == 'GET' else request.form
_is_ok, _error = User.user_update(params)
print _is_ok
print _error
return jsonify({'is_ok': _is_ok, 'error': _error})
from modules.database import Database
from modules.user import User
Database.initialize()
User.update_user_email("*****@*****.**", "*****@*****.**")
