def like_photo(photo_id):
global current_page
if 'username' not in session:
return redirect('/login')
nm = PhotoModel(db.get_connection())
nm.likeit(photo_id)
return redirect("/" + current_page)
def index():
global current_page
current_page = 'index'
if 'username' not in session:
return redirect('/login')
photo = PhotoModel(db.get_connection()).get_all(session['user_id'])
return render_template('index.html',
username=session['username'],
photo=photo)
def user(user_id):
global current_page
current_page = 'user/' + str(user_id)
photo = PhotoModel(db.get_connection()).get_all(user_id)
user = [UsersModel(db.get_connection()).get(user_id)]
return render_template('user.html',
username=session['username'],
photo=photo,
user=user)
def add_photo():
global current_page
if 'username' not in session:
return redirect('/login')
form = AddPhotoForm()
if form.validate_on_submit():
title = form.title.data
content = form.content.data
filename = secure_filename(content.filename)
if os.path.isfile(os.path.join('static', 'img', filename)):
while os.path.isfile(os.path.join('static', 'img', filename)):
filename = filename.split('.')
filename = '.'.join([filename[0] + 'A', filename[-1]])
content.save(os.path.join('static', 'img', filename))
nm = PhotoModel(db.get_connection())
nm.insert(title, filename, session['user_id'], 0, session['username'])
return redirect("/" + current_page)
return render_template('add_photo.html',
title='Добавить фото',
form=form,
username=session['username'])
def posted_photo(photo_id):
global current_page
current_page = 'photo/' + str(photo_id)
photo = [PhotoModel(db.get_connection()).get(photo_id)]
comments = CommentModel(db.get_connection()).get(photo_id)
form = AddCommentForm()
if form.validate_on_submit():
CommentText = form.CommentText.data
nm = CommentModel(db.get_connection())
nm.insert(session['username'], photo_id, CommentText)
return redirect('/photo/' + str(photo_id))
if 'username' in session:
return render_template('photo.html',
photo=photo,
comments=comments,
form=form,
username=session['username'])
else:
return render_template('photo.html',
photo=photo,
comments=comments,
form=form,
username='******')
def main():
global current_page
current_page = 'main'
photo = PhotoModel(db.get_connection()).get_all()
return render_template('main.html', photo=photo)
def delete_photo(photo_id):
if 'username' not in session:
return redirect('/login')
nm = PhotoModel(db.get_connection())
nm.delete(photo_id)
return redirect("/index")
from db import DB
from flask import Flask, redirect, render_template, session
from loginform import LoginForm
from regform import RegForm
from photo_model import PhotoModel
from users_model import UsersModel
from comment_model import CommentModel
from add_comment import AddCommentForm
from werkzeug.utils import secure_filename
import os
app = Flask(__name__)
app.config['SECRET_KEY'] = 'jOaqY9515WL6IxQB'
# инициализируем таблицу
db = DB('db.db')
PhotoModel(db.get_connection()).init_table()
UsersModel(db.get_connection()).init_table()
CommentModel(db.get_connection()).init_table()
current_page = 'main'
# http://127.0.0.1:8080/login
@app.route('/login', methods=['GET', 'POST'])
def login():
form = LoginForm()
if form.validate_on_submit():
user_name = form.username.data
password = form.password.data
user_model = UsersModel(db.get_connection())
exists = user_model.exists(user_name, password)
if (exists[0]):