def add_dealer():
if 'username' not in session:
return redirect('/login')
if session['username'] == 'admin':
form = AddDealerForm()
if form.validate_on_submit():
dealers = DealersModel(db.get_connection())
dealers.insert(name=form.name.data, address=form.address.data)
return redirect(url_for('index'))
return render_template("add_dealer.html",
title='Adding a dealership',
form=form)
def del_dealer(dealer_id):
"""
Удаление дилера
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
return redirect('index')
dealer = DealersModel(db.get_connection())
dealer.delete(dealer_id)
return redirect("/index")
def add_car():
"""
Добавление модели обуви
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
return redirect('index')
form = AddCarForm()
available_dealers = [(i[0], i[1])
for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
# создать автомобиль
cars = CarsModel(db.get_connection())
cars.insert(model=form.model.data,
price=form.price.data,
power=form.power.data,
color=form.color.data,
dealer=form.dealer_id.data)
# редирект на главную страницу
return redirect(url_for('car_admin'))
return render_template("add_car.html", title='Добавление обуви', form=form)
def add_dealer():
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если админ, то его на свою страницу
if session['username'] == 'admin':
form = AddDealerForm()
if form.validate_on_submit():
# создать дилера
dealers = DealersModel(db.get_connection())
dealers.insert(name=form.name.data, address=form.address.data)
# редирект на главную страницу
return redirect(url_for('index'))
return render_template("add_dealer.html",
title='Добавление магазина',
form=form)
def add_car():
"""
Добавление автомобиля
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
return redirect('index')
form = AddCarForm()
available_dealers = [(i[0], i[1]) for i in DealersModel(db.get_connection()).get_all()]
form.country.choices = available_dealers
if form.validate_on_submit():
# создать автомобиль
form.save_im()
print(form.image, form.image.data)
cars = CarsModel(db.get_connection())
cars.insert(model=form.name.data,
price=form.price.data,
power=form.adress.data,
color=form.description.data,
dealer=form.country.data,
image=os.path.join('static', 'img', form.image.data.filename))
# редирект на главную страницу
return redirect(url_for('car_admin'))
return render_template("add_car.html", title='Добавление отеля', form=form)
def delete_breed(car_id):
if 'username' not in session:
return redirect('/login')
if session['username'] != 'admin':
return redirect(url_for('index'))
DealersModel(db.get_connection()).delete(car_id)
return redirect(url_for('breed_admin'))
def add_horse():
"""
Добавление автомобиля
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
return redirect('index')
form = AddCarForm()
available_dealers = [(i[0], i[1]) for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
# создать автомобиль
cars = CarsModel(db.get_connection())
#file = form.image.data
#print(form.image.data)
#file.save(os.path.join(
# "static", "img", file.filename
# ))
filename = os.path.join(
"static", "img", "horse.jpg"
)
cars.insert(model=form.model.data,
price=form.price.data,
power=form.power.data,
image_id=filename,
dealer=form.dealer_id.data)
# редирект на главную страницу
return redirect(url_for('horse_admin'))
return render_template("add_car.html", title='Добавление автомобиля', form=form)
def dealer(dealer_id):
if 'username' not in session:
return redirect('/login')
if session['username'] != 'admin':
return redirect(url_for('index'))
dealer = DealersModel(db.get_connection()).get(dealer_id)
return render_template('dealer_info.html',
username=session['username'],
title='View dealership information',
dealer=dealer)
def motorcycle(motorcycle_id):
if 'username' not in session:
return redirect('/login')
motorcycle = MotorcycleModel(db.get_connection()).get(motorcycle_id)
dealer = DealersModel(db.get_connection()).get(motorcycle[5])
return render_template('motorcycle_info.html',
username=session['username'],
title='View Motorcycle',
motorcycle=motorcycle,
dealer=dealer[1])
def dealer(dealer_id):
if 'username' not in session:
return redirect('/login')
if session['username'] != 'admin':
return redirect(url_for('index'))
dealer = DealersModel(db.get_connection()).get(dealer_id)
return render_template('dealer_info.html',
username=session['username'],
title='Просмотр информации о продавце',
dealer=dealer)
def doll(doll_id):
if 'username' not in session:
return redirect('/login')
doll = DollsModel(db.get_connection()).get(doll_id)
dealer = DealersModel(db.get_connection()).get(doll[5])
return render_template('doll_info.html',
username=session['username'],
title='Просмотр кукол',
doll=doll,
dealer=dealer[1])
def dealer_admin():
if 'username' not in session:
return redirect('/login')
if session['username'] != 'admin':
flash('Access is denied')
redirect('index')
dealers = DealersModel(db.get_connection()).get_all()
return render_template('dealer_admin.html',
username=session['username'],
title='View dealerships',
dealers=dealers)
def dealer_admin():
if 'username' not in session:
return redirect('/login')
if session['username'] != 'admin':
flash('Доступ запрещен')
redirect('index')
dealers = DealersModel(db.get_connection()).get_all()
return render_template('dealer_admin.html',
username=session['username'],
title='Просмотр продавцов',
dealers=dealers)
def search_dealer():
form = SearchDealerForm()
available_dealers = [(i[0], i[1]) for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
#
cars = CarsModel(db.get_connection()).get_by_dealer(form.dealer_id.data)
# редирект на главную страницу
return render_template('car_user.html', username=session['username'], title='Просмотр базы', cars=cars)
return render_template("search_dealer.html", title='Подбор по цене', form=form)
def car(car_id):
if 'username' not in session:
return redirect('/login')
'''if session['username'] != 'admin':
return redirect(url_for('index'))'''
car = CarsModel(db.get_connection()).get(car_id)
dealer = DealersModel(db.get_connection()).get(car[5])
return render_template('car_info.html',
username=session['username'],
title='Просмотр дома',
car=car,
dealer=dealer[1])
def search_dealer():
"""
Запрос автомобилей, продающихся в определенном дилерском центре
"""
form = SearchDealerForm()
available_dealers = [(i[0], i[1]) for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
#
books = BooksModel(db.get_connection()).get_by_dealer(form.dealer_id.data)
# редирект на главную страницу
return render_template('book_user.html', username=session['username'], title='Просмотр базы', books=books)
return render_template("search_dealer.html", title='Подбор по цене', form=form)
def dealer_():
"""
Вывод всей информации об всех дилерских центрах
:return:
информация для авторизованного пользователя
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
dealers = DealersModel(db.get_connection()).get_all()
return render_template('dealers.html',
username=session['username'],
title='Просмотр наших размещений',
dealers=dealers)
def search_dealer():
form = SearchDealerForm()
available_dealers = [(i[0], i[1])
for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
dolls = DollsModel(db.get_connection()).get_by_dealer(
form.dealer_id.data)
return render_template('doll_user.html',
username=session['username'],
title='Просмотр базы',
dolls=dolls)
return render_template("search_dealer.html",
title='Подбор по цене',
form=form)
def dealer(dealer_id):
"""
Вывод всей информации о дилерском центре
:return:
информация для авторизованного пользователя
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# иначе выдаем информацию
dealer = DealersModel(db.get_connection()).get(dealer_id)
return render_template('dealer_info.html',
username=session['username'],
title='Просмотр информации о наших местоположениях',
dealer=dealer)
def search_dealer():
form = SearchDealerForm()
available_dealers = [(i[0], i[1])
for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
motorcycles = MotorcycleModel(db.get_connection()).get_by_dealer(
form.dealer_id.data)
return render_template('motorcycle_user.html',
username=session['username'],
title='View database',
motorcycles=motorcycles)
return render_template("search_dealer.html",
title='Pricing by price',
form=form)
def dealer_admin():
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
flash('Доступ запрещен')
redirect('index')
# иначе это админ
dealers = DealersModel(db.get_connection()).get_all()
return render_template('dealer_admin.html',
username=session['username'],
title='Просмотр по магазинам',
dealers=dealers)
def car(car_id):
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если не админ, то его на главную страницу
'''if session['username'] != 'admin':
return redirect(url_for('index'))'''
# иначе выдаем информацию
car = CarsModel(db.get_connection()).get(car_id)
dealer = DealersModel(db.get_connection()).get(car[5])
return render_template('car_info.html',
username=session['username'],
title='Просмотр игры',
car=car,
dealer=dealer[1])
def dealer(dealer_id):
"""
Вывод всей информации о магазине
:return:
информация для авторизованного пользователя
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если не админ, то его на главную страницу
if session['username'] != 'admin':
return redirect(url_for('index'))
# иначе выдаем информацию
dealer = DealersModel(db.get_connection()).get(dealer_id)
return render_template('dealer_info.html',
username=session['username'],
title='Просмотр информации о магазине',
dealer=dealer)
def add_car():
if 'username' not in session:
return redirect('login')
if session['username'] != 'admin':
return redirect('index')
form = AddCarForm()
available_dealers = [(i[0], i[1])
for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
cars = CarsModel(db.get_connection())
cars.insert(model=form.model.data,
price=form.price.data,
power=form.power.data,
color=form.color.data,
dealer=form.dealer_id.data)
return redirect(url_for('car_admin'))
return render_template("add_car.html", title='Добавление дома', form=form)
def add_doll():
if 'username' not in session:
return redirect('login')
if session['username'] != 'admin':
return redirect('index')
form = AddDollForm()
available_dealers = [(i[0], i[1])
for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
dolls = DollsModel(db.get_connection())
dolls.insert(price=form.price.data,
color_of_hair=form.color_of_hair.data,
rarity=form.rarity.data,
dealer_id=form.dealer_id.data)
return redirect(url_for('doll_admin'))
return render_template("add_doll.html",
title='Добавление куклы',
form=form)
def dealer_admin():
"""
Вывод всей информации об всех дилерских центрах
:return:
информация для авторизованного пользователя
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если админ, то его на свою страницу
if session['username'] != 'admin':
flash('Доступ запрещен')
redirect('index')
# иначе это админ
dealers = DealersModel(db.get_connection()).get_all()
return render_template('dealer_admin.html',
username=session['username'],
title='Просмотр магазинов',
dealers=dealers)
def book(book_id):
"""
Вывод всей информации об автомобиле
:return:
информация для авторизованного пользователя
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если не админ, то его на главную страницу
'''if session['username'] != 'admin':
return redirect(url_for('index'))'''
# иначе выдаем информацию
book = BooksModel(db.get_connection()).get(book_id)
dealer = DealersModel(db.get_connection()).get(book[5])
return render_template('book_info.html',
username=session['username'],
title='Просмотр книг',
book=book,
dealer=dealer[1])
def add_motorcycle():
if 'username' not in session:
return redirect('login')
if session['username'] != 'admin':
return redirect('index')
form = AddMotorcycleForm()
available_dealers = [(i[0], i[1])
for i in DealersModel(db.get_connection()).get_all()]
form.dealer_id.choices = available_dealers
if form.validate_on_submit():
motorcycles = MotorcycleModel(db.get_connection())
motorcycles.insert(model=form.model.data,
price=form.price.data,
power=form.power.data,
color=form.color.data,
dealer=form.dealer_id.data)
return redirect(url_for('motorcycle_admin'))
return render_template("add_motorcycle.html",
title='Add motorcycle',
form=form)
def car(car_id):
"""
Вывод всей информации об автомобиле
:return:
информация для авторизованного пользователя
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если не админ, то его на главную страницу
'''if session['username'] != 'admin':
return redirect(url_for('index'))'''
# иначе выдаем информацию
car = CarsModel(db.get_connection()).get(car_id)
dealer = DealersModel(db.get_connection()).get(car[5])
isAdmin = session['username'] == "admin"
print(session['username'] == "admin", session['username'])
return render_template('car_info.html',
isAdmin=isAdmin,
car_id=car_id,
username=session['username'],
title='Просмотр автомобиля',
car=car,
dealer=dealer[1])
from flask import Flask, session, redirect, render_template, flash, url_for
from werkzeug.security import generate_password_hash, check_password_hash
from models import UsersModel, CarsModel, DealersModel
from forms import LoginForm, RegisterForm, AddCarForm, SearchPriceForm, SearchDealerForm, AddDealerForm
from db import DB
app = Flask(__name__)
app.config['SECRET_KEY'] = 'yandexlyceum_secret_key'
db = DB()
UsersModel(db.get_connection()).init_table()
CarsModel(db.get_connection()).init_table()
DealersModel(db.get_connection()).init_table()
@app.route('/')
@app.route('/index')
def index():
"""
Главная страница
:return:
Основная страница сайта, либо редирект на авторизацю
"""
# если пользователь не авторизован, кидаем его на страницу входа
if 'username' not in session:
return redirect('/login')
# если админ, то его на свою страницу
if session['username'] == 'admin':
return render_template('index_admin.html',
username=session['username'])
# если обычный пользователь, то его на свою
cars = CarsModel(db.get_connection()).get_all()
