def profile(usern):
if not db.user_exists(usern):
return render_template('404.html', obj="user")
originals = [db.get_title(sid) for sid in db.stories_by_user(usern)]
contributed = [(len(db.contributions_to_story(usern, sid)), db.get_title(sid)) for sid in db.stories_with_user_contributions(usern)]
contributed.sort(reverse=True)
return render_template('user.html', username=usern, originals=originals, contribs=contributed)
def create_user():
if request.method == 'POST':
if db.user_exists(request.form['username']):
flash('User already exsits')
elif request.form['password'] != request.form['password2']:
flash('Passwords do not match')
else:
collection.insert_one({
'first_name':
request.form['first_name'],
'last_name':
request.form['last_name'],
'username':
request.form['username'],
'password':
enc.encrypt(request.form['password']),
'date_created':
int(time())
})
session['USERNAME'] = request.form['username']
session['PASSWORD'] = request.form['password']
return redirect(url_for('index'))
else:
return 'Please use a post request'
def change_username():
auth_status = auth()
if not auth_status[0]:
return redirect(url_for('index'))
if request.method == 'GET':
return redirect(url_for('settings.index'))
if user_exists(request.form['username']):
flash(f'Username: {request.form["username"]} is already in use')
return redirect(url_for('settings.index'))
if not authenticate(auth_status[1].username, request.form['password']):
flash('Incorrect current password, your username has NOT changed')
return redirect(url_for('settings.index'))
collection.update_one(
{'_id': auth_status[1].id},
{ '$set': {
'username': request.form['username']
}}
)
session['USERNAME'] = request.form['username']
flash(f'Username changed from {auth_status[1].username} to {request.form["username"]}')
return redirect(url_for('settings.index'))
def get_user():
logger.debug("Getting user")
user_id = request.cookies.get('user-id')
logging.info("User id from cookie: %s", user_id)
user_exists = db.user_exists(user_id) if user_id else False
logging.info("User exists? %s", user_exists)
return User(id=user_id) if user_exists else None
def register_user(username, password):
"""helper function to register a user"""
if db.user_exists(username):
return "User already exists", 409
hashed_pass = generate_password_hash(password)
db.create_user(username, hashed_pass)
return "User Created", 200
def login_post(response):
username = response.get_field("username")
user_exists = db.user_exists(username)
if user_exists:
player_id = db.get_player_id(username)
login_cookies(response, username, player_id)
else:
response.redirect("/")
def login_post(response):
username = response.get_field("username")
user_exists = db.user_exists(username)
if user_exists:
player_id = db.get_player_id(username)
login_cookies(response, username, player_id)
else:
response.redirect("/")
def login():
uname = session.get("username")
if uname and user_exists(uname):
return redirect("/")
err = None
if request.method == "POST":
username = request.form.get("username", None)
password = request.form.get("password", None)
ok, err = check_user(username, password)
if ok:
session["username"] = username
return redirect("/")
return render_template("login.html", err=err)
def start_message(message):
"""Greeting of useer when he first time in bot.
And also create entry in database for his information
Args:
message: answer from telegram api.
"""
chat_id = message.chat.id
bot.send_message(
message.chat.id,
messages.intro_message,
reply_markup=keyboard_main,
)
if not db.user_exists(chat_id):
db.new_user(chat_id, message.from_user.username)
def register_user():
error = None
username = request.form['username']
password = request.form['password']
first_name = request.form.get('first_name')
last_name = request.form.get('last_name')
email = request.form.get('email')
isEmployer = request.form.get('isEmployer')
if request.method != 'POST':
error = "Invalid request method"
if not username:
error = "Username is required"
elif not password:
error = "Password is required"
elif db.user_exists(username):
error = "Username {} is already registered".format(username)
if error is None:
db.create_user(username, password, first_name, last_name, email,
isEmployer)
resp = {
"status": "success",
"message": "Successfully registered user",
"username": username
}
user = db.get_user_by_username(username)
session['user_id'] = user['user_id']
session['username'] = username
return jsonify(resp)
resp = {
"status": "failure",
"messsage": "unable to regsiter user",
"reason": error
}
return jsonify(resp)
def add_offer(message, shop, item):
"""add dungling offer that user need to approve
Args:
message (): bounty
shop (str): shop for offer
item (str): item for offer
"""
chat_id = message.chat.id
bounty = message.text
# Если юзер не прожимал старт во время запуска бота(хз как) создаем его док в базе
if db.user_exists(chat_id):
db.add_dungling_offer(chat_id, shop, item, bounty)
else:
db.new_user(chat_id, message.from_user.username)
db.add_dungling_offer(chat_id, shop, item, bounty)
bot.send_message(chat_id,
messages.print_offer.format(shop, item, bounty),
reply_markup=keyboard_confirmation)
def test_create_exists_user():
create_user('test', 'test_pass')
assert user_exists('test') is True
def test_exists_user_negative():
assert user_exists('noexist') is False
def input_existing_user():
while True:
name = input("User: ")
if db.user_exists(name):
break
return name
def decorated_func(*args, **kwargs):
uname = session.get("username")
if not uname or not user_exists(uname):
return redirect(url_for("views.login"))
return f(*args, **kwargs)
def decorated_func(*args, **kwargs):
uname = args[0].session.get('username')
if not uname or not user_exists(session, uname)[0]:
return redirect('/login')
return f(*args, **kwargs)
def register_user_post(response):
username = response.get_field("username")
if not db.user_exists(username):
player_id = db.add_player(username)
login_cookies(response, username, player_id)
def register_user_post(response):
username = response.get_field("username")
if not db.user_exists(username):
player_id = db.add_player(username)
login_cookies(response, username, player_id)
def validate_user(form, field):
if (db.user_exists(field.data)):
raise ValidationError("This username is already in use!")
