def check_registration():
user = User.get(login='******')
if user is None:
user = User(login='******', vk_id=1)
user.save('123456')
print(user.check_user('123456'))
print(user.vk_user)
def match(form, field):
user = None
if cur_user():
user = cur_user()
elif form.login.data is not '':
user = User.get(login=form.login.data)
if user and not user.check_pass(field.data):
raise ValidationError("Неправильный пароль")
def cabinet(usr, tab=0):
"""
Отвечает за вывод страницы личного кабинета
:return: Страница личного кабинета
"""
video_list = Video.get()
user = cur_user()
cabinet_owner = User.get(login=usr)
is_cabinet_settings_available = False
if user == cabinet_owner:
is_cabinet_settings_available = True
items = cabinet_owner.videos
form = UserProfileForm()
form_acc = AccountSettingsForm()
if request.method == 'POST':
form_name = request.form['form-name']
tab = 2
if form_name == 'form':
tab = 1
if form_name == 'form' and form.validate():
tab = 1
user = cur_user()
folder = str(user.id)
if form.change_name.data:
user.change_name(form.change_name.data)
if form.channel_info.data:
user.change_channel_info(form.channel_info.data)
if 'avatar' in request.files:
avatar_url = avatars.save(form.avatar.data, folder=folder)
user.update_avatar(json.dumps({"url": avatar_url}))
if 'background' in request.files:
background_url = backgrounds.save(form.background.data,
folder=folder)
user.update_background(json.dumps({"url": background_url}))
return redirect(
url_for("cabinet", usr=cabinet_owner.login, tab=tab))
elif form_name == 'form_acc' and form_acc.validate():
tab = 2
user = cur_user()
if form_acc.change_password.data:
user.save(form_acc.change_password.data)
return redirect(
url_for("cabinet", usr=cabinet_owner.login, tab=tab))
now = time = datetime.now(tz=None)
return render_template('user/cabinet.html',
form=form,
form_acc=form_acc,
user=user,
items=items,
settings=is_cabinet_settings_available,
usr=cabinet_owner,
subscribed=(user in cabinet_owner.subscribers),
now=now,
tab=tab)
def check_information():
author = User.get(login='******').vk_user
unexp_inf = Information(
author_id=author.vk_id,
text='This is the most valuable information in the world.')
unexp_inf.save()
exp_inf = Information(
author_id=author.vk_id,
text='This is the most valuable information in the world for now.',
expiration_time=datetime.utcnow() + timedelta(0, 0, 2))
exp_inf.save()
def parse_signed_cookie(cookie_str):
try:
L = cookie_str.split('-')
if len(L) != 3:
return None
id, expires, md5 = L
if int(expires) < time.time():
return None
user = User.get(id)
if user is None:
return None
if md5 != hashlib.md5('%s-%s-%s-%s' % (id, user.password, expires, _COOKIE_KEY)).hexdigest():
return None
return user
except:
return None
def play(vid):
video = Video.get(video_id=vid)
if not video:
abort = Aborter()
return abort(404)
user = cur_user()
usr = User.get(login=video.user.login)
if user and user not in video.viewers:
video.add_viewer(user)
likened = 0
if user in video.likes:
likened = 1
if user in video.dislikes:
likened = -1
return render_template('video_page.html',
user=user,
vid=vid,
video=video,
lkd=likened,
usr=usr,
subscribed=(user in usr.subscribers))
def not_exist(form, field):
if User.get(login=field.data) is None:
raise ValidationError("Такого пользователя не существует")
def exist(form, field):
if User.get(login=field.data):
raise ValidationError("Такой пользователь уже существует")
def cur_user():
if 'Login' in session:
return User.get(login=session['Login'])
return None
def test_is_get_method_work(self):
self.assertEqual(User.get(self.user.id), self.user)
def check_auth_password(form, field):
user = User.get(login=form.login.data)
if user is None:
pass
elif not user.check_user(field.data):
raise ValidationError('Неправильный пароль')
def check_auth_login(form, field):
user = User.get(login=field.data)
if user is None:
raise StopValidation('Неизвестный логин.')
def validate_login_is_uniq(form, field):
user = User.get(login=field.data)
if user is not None:
raise ValidationError('Пользователь с таким логином уже существует.')
def validate_vk_id_is_uniq(form, field):
vk_id = get_numeric_id(field.data)
if User.get(vk_id=vk_id) is not None:
raise ValidationError('Пользователь с таким ID уже зарегистрирован.')
