259 lines
9.4 KiB
Python
259 lines
9.4 KiB
Python
import datetime
|
|
from random import randint
|
|
|
|
from flask import Flask, render_template
|
|
from flask_login import LoginManager, login_user, logout_user, login_required, current_user
|
|
from werkzeug.utils import redirect
|
|
|
|
from data import db_session
|
|
from data.diary_post import DiaryPost
|
|
from data.users import User
|
|
from forms.login import LoginForm
|
|
from forms.post import AddPost
|
|
from forms.recovery import RecoveryForm, Conf, Finish
|
|
from forms.register import RegisterForm, Confirmation
|
|
from post import mail
|
|
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = 'moona_secret_key'
|
|
login_manager = LoginManager()
|
|
login_manager.init_app(app)
|
|
help_arg = False
|
|
send_msg = False
|
|
secret_code = None
|
|
photo = None
|
|
user_email = ""
|
|
|
|
|
|
def norm_data(datatime, date_or_time, r=False):
|
|
if date_or_time == 'date':
|
|
return '.'.join(str(datatime).split()[0].split('-')[::-1])
|
|
elif date_or_time == 'time':
|
|
return ':'.join(str(datatime).split()[1].split(':')[0:2])
|
|
elif date_or_time == 'datetime':
|
|
date = '.'.join(str(datatime).split()[0].split('-')[::-1])
|
|
times = ':'.join(str(datatime).split()[1].split(':')[0:2])
|
|
datatimes = date + ' ' + times if r else times + ' ' + date
|
|
datatimes = datetime
|
|
return datatimes
|
|
|
|
|
|
def save_photo(photo, login, post=False, id_post=None):
|
|
if not post:
|
|
with open(f'static/app_image/users_photo/{login}_logo.png', 'wb') as f:
|
|
photo.save(f)
|
|
return f'../static/app_image/users_photo/{login}_logo.png'
|
|
elif post and id_post != None:
|
|
with open(f'static/app_image/post_photo/{login}_post_{id_post}.png', 'wb') as f:
|
|
photo.save(f)
|
|
return f'../static/app_image/post_photo/{login}_post_{id_post}.png'
|
|
|
|
|
|
def secret_key():
|
|
return ''.join([str(randint(0, 9)) for i in range(5)])
|
|
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
db_sess = db_session.create_session()
|
|
return db_sess.query(User).get(user_id)
|
|
|
|
|
|
@app.route('/')
|
|
def main_page():
|
|
return render_template('base.html', title='moona')
|
|
|
|
|
|
@app.route('/add_post', methods=['GET', 'POST'])
|
|
def add_post():
|
|
pos = AddPost()
|
|
session = db_session.create_session()
|
|
if pos.validate_on_submit():
|
|
id = session.query(DiaryPost).order_by(DiaryPost.id)[-1].id
|
|
if id:
|
|
id += 1
|
|
else:
|
|
id = -1
|
|
if pos.photo.data:
|
|
diart_pos = DiaryPost(name=pos.name.data,
|
|
text=pos.text.data,
|
|
author=current_user.id,
|
|
date=datetime.datetime.now(),
|
|
photo=save_photo(pos.photo.data, current_user.login, post=True, id_post=id),
|
|
public=pos.public.data,
|
|
pos_emot=pos.pos_emot.data,
|
|
nig_emot=pos.nig_emot.data,
|
|
link=pos.link.data)
|
|
session.add(diart_pos)
|
|
session.commit()
|
|
return redirect("/diary")
|
|
else:
|
|
diart_pos = DiaryPost(name=pos.name.data,
|
|
text=pos.text.data,
|
|
author=current_user.id,
|
|
date=datetime.datetime.now(),
|
|
public=pos.public.data,
|
|
pos_emot=pos.pos_emot.data,
|
|
nig_emot=pos.nig_emot.data,
|
|
link=pos.link.data)
|
|
session.add(diart_pos)
|
|
session.commit()
|
|
return redirect("/diary")
|
|
return render_template('post.html', form=pos)
|
|
|
|
|
|
@app.route('/diary', methods=['GET', 'POST'])
|
|
def diary():
|
|
db_sess = db_session.create_session()
|
|
if current_user.is_authenticated:
|
|
posts = db_sess.query(DiaryPost).filter(DiaryPost.author == current_user.id).all()
|
|
else:
|
|
posts = None
|
|
return render_template('diary.html', title='moona', post=posts)
|
|
|
|
|
|
@app.route('/logout')
|
|
@login_required
|
|
def logout():
|
|
logout_user()
|
|
return redirect("/")
|
|
|
|
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
db_sess = db_session.create_session()
|
|
user = db_sess.query(User).filter(User.email == form.email.data).first()
|
|
if user and user.check_password(form.password.data):
|
|
login_user(user, remember=form.remember_me.data)
|
|
return redirect("/")
|
|
return render_template('login.html',
|
|
message="Неправильный логин или пароль",
|
|
form=form)
|
|
return render_template('login.html', title='Авторизация', form=form, message='')
|
|
|
|
|
|
@app.route('/confirmation', methods=['GET', 'POST'])
|
|
def confirmation():
|
|
global help_arg
|
|
global send_msg
|
|
global secret_code
|
|
global photo
|
|
form = help_arg
|
|
session = db_session.create_session()
|
|
conf = Confirmation()
|
|
if not send_msg:
|
|
secret_code = secret_key()
|
|
mail(f'Ваш секретный код: {secret_code}', form.email.data, 'Moona Код')
|
|
print(secret_code)
|
|
send_msg = True
|
|
if conf.validate_on_submit():
|
|
if str(conf.code_key.data).strip() == str(secret_code).strip():
|
|
print(secret_code)
|
|
if form.photo.data:
|
|
user = User(
|
|
name=form.name.data,
|
|
surname=form.surname.data,
|
|
login=form.login.data,
|
|
age=form.age.data,
|
|
about=form.about.data,
|
|
email=form.email.data,
|
|
photo=photo,
|
|
role='user'
|
|
)
|
|
else:
|
|
user = User(
|
|
name=form.name.data,
|
|
surname=form.surname.data,
|
|
login=form.login.data,
|
|
age=form.age.data,
|
|
about=form.about.data,
|
|
email=form.email.data,
|
|
role='user'
|
|
)
|
|
user.set_password(form.password.data)
|
|
session.add(user)
|
|
session.commit()
|
|
send_msg = False
|
|
return redirect('/login')
|
|
else:
|
|
return render_template('confirmation_reg.html', title='Подтверждение', form=conf,
|
|
message='Коды не совпадают')
|
|
return render_template('confirmation_reg.html', title='Подтверждение', form=conf, message='')
|
|
|
|
|
|
@app.route('/register', methods=['GET', 'POST'])
|
|
def register():
|
|
global help_arg
|
|
global photo
|
|
form = RegisterForm()
|
|
if form.validate_on_submit():
|
|
if form.password.data != form.password2.data:
|
|
return render_template('register.html', title='Регистрация',
|
|
form=form,
|
|
message="Пароли не совпадают")
|
|
session = db_session.create_session()
|
|
if session.query(User).filter(User.login == form.login.data).first():
|
|
return render_template('register.html', title='Регистрация',
|
|
form=form,
|
|
message="Такой пользователь уже есть")
|
|
help_arg = form
|
|
if form.photo.data:
|
|
photo = save_photo(form.photo.data, form.login.data)
|
|
return redirect('/confirmation')
|
|
return render_template('register.html', title='Регистрация', form=form, message='')
|
|
|
|
|
|
@app.route('/recovery', methods=['GET', 'POST'])
|
|
def recovery():
|
|
global send_msg
|
|
global secret_code
|
|
global help_arg
|
|
global user_email
|
|
form = RecoveryForm()
|
|
conf = Conf()
|
|
finish = Finish()
|
|
session = db_session.create_session()
|
|
if form.validate_on_submit() and form.email.data:
|
|
user_email = form.email.data
|
|
if not send_msg:
|
|
secret_code = secret_key()
|
|
mail(f'Ваш секретный код: {secret_code}', form.email.data, 'Moona Код')
|
|
send_msg = True
|
|
print(secret_code)
|
|
return render_template('recovery.html', title='Восстановление пароля', form=conf, message='', s='2')
|
|
if conf.validate_on_submit():
|
|
if str(conf.code_key.data).strip() == str(secret_code).strip():
|
|
help_arg = True
|
|
return render_template('recovery.html', title='Восстановление пароля', form=finish, message='', s='3')
|
|
if help_arg:
|
|
if finish.validate_on_submit():
|
|
db_sess = db_session.create_session()
|
|
user = db_sess.query(User).filter(User.email == user_email).first()
|
|
# user2 = User(
|
|
# name=user.name,
|
|
# surname=user.surname,
|
|
# login=user.login,
|
|
# age=user.age,
|
|
# about=user.about,
|
|
# email=user_email,
|
|
# photo=user.photo,
|
|
# role='user')
|
|
user.set_password(finish.password.data)
|
|
user2 = session.merge(user)
|
|
session.add(user2)
|
|
session.commit()
|
|
send_msg = False
|
|
return redirect('/login')
|
|
return render_template('recovery.html', title='Восстановление пароля', form=form, message='', s='1')
|
|
|
|
|
|
def main():
|
|
db_session.global_init("db/moona_data.db")
|
|
app.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|