Рефакторинг
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
from flask import request, Blueprint, render_template, session, redirect, url_for
|
from flask import request, Blueprint, render_template, session, redirect, url_for
|
||||||
from os import path
|
from os import path
|
||||||
from db.sql_provider import SQLProvider
|
from Database.sql_provider import SQLProvider
|
||||||
from db.select import select_list
|
from Database.select import select_list
|
||||||
import json
|
import json
|
||||||
|
|
||||||
sql_provider = SQLProvider(path.join(path.dirname(__file__), 'sql'))
|
sql_provider = SQLProvider(path.join(path.dirname(__file__), 'sql'))
|
||||||
@@ -15,14 +15,17 @@ def auth():
|
|||||||
if request.method == 'GET':
|
if request.method == 'GET':
|
||||||
return render_template('auth.html')
|
return render_template('auth.html')
|
||||||
else:
|
else:
|
||||||
data = dict(request.form)
|
data = request.form.to_dict()
|
||||||
print(data)
|
print(data)
|
||||||
result = select_list(config, sql_provider.get('auth.sql', data))
|
result = select_list(config, sql_provider.get('auth.sql', data))
|
||||||
|
|
||||||
if result[1]:
|
if result[1]:
|
||||||
session['user_id'] = result[1][0]['user_ID']
|
session.update({
|
||||||
session['role'] = result[1][0]['role']
|
'user_id': result[1][0]['user_ID'],
|
||||||
session['db_config'] = config
|
'role': result[1][0]['role'],
|
||||||
session.permanent = True
|
'db_config': config,
|
||||||
|
'permanent': True
|
||||||
|
})
|
||||||
return redirect(url_for('index'))
|
return redirect(url_for('index'))
|
||||||
else:
|
|
||||||
return 'Неправильный логин или пароль'
|
return 'Неправильный логин или пароль'
|
||||||
|
|||||||
0
App/Database/__init__.py
Normal file
0
App/Database/__init__.py
Normal file
@@ -1,13 +0,0 @@
|
|||||||
import db.DBconnect
|
|
||||||
|
|
||||||
sql_provider = 1
|
|
||||||
""" def select_list(db_config, sql):
|
|
||||||
with DBContextManager(db_config) as cursor:
|
|
||||||
if cursor is None:
|
|
||||||
raise ValueError("Cursor not created")
|
|
||||||
else:
|
|
||||||
cursor.execute(sql)
|
|
||||||
result = cursor.fetchall()
|
|
||||||
schema = [item[0] for item in cursor.description]
|
|
||||||
lst = [dict(zip(schema, row)) for row in result]
|
|
||||||
return schema, lst """
|
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
from flask import request, Blueprint, render_template, session, redirect, url_for
|
from flask import request, Blueprint, render_template, session
|
||||||
from os import path
|
from os import path
|
||||||
from db.sql_provider import SQLProvider
|
from Database.sql_provider import SQLProvider
|
||||||
from db.select import select_list
|
from Database.select import select_list
|
||||||
from checker import check_auth
|
from checker import check_auth
|
||||||
|
|
||||||
sql_provider = SQLProvider(path.join(path.dirname(__file__), 'sql'))
|
sql_provider = SQLProvider(path.join(path.dirname(__file__), 'sql'))
|
||||||
@@ -15,6 +15,5 @@ def sklad_zapros():
|
|||||||
return render_template('sklad_zapros.html', materials=result[1], status=True)
|
return render_template('sklad_zapros.html', materials=result[1], status=True)
|
||||||
else:
|
else:
|
||||||
name = dict(request.form)
|
name = dict(request.form)
|
||||||
print(name)
|
|
||||||
result = select_list(session['db_config'], sql_provider.get('zapros1.sql', name))
|
result = select_list(session['db_config'], sql_provider.get('zapros1.sql', name))
|
||||||
return render_template('sklad_zapros.html', materials=result[1], result_table=result)
|
return render_template('sklad_zapros.html', materials=result[1], result_table=result)
|
||||||
@@ -1,11 +1,10 @@
|
|||||||
from flask import Flask, redirect, render_template, url_for, session, request
|
from flask import Flask, render_template, session
|
||||||
from os import path
|
|
||||||
from Queries.req import requests_bp
|
from Queries.req import requests_bp
|
||||||
from Auth.auth_route import auth_bp
|
from Auth.auth_route import auth_bp
|
||||||
from checker import check_auth
|
from checker import check_auth
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
app.secret_key = 'super secret key'
|
app.secret_key = 'suplex'
|
||||||
|
|
||||||
app.register_blueprint(requests_bp, url_prefix='/requests')
|
app.register_blueprint(requests_bp, url_prefix='/requests')
|
||||||
app.register_blueprint(auth_bp, url_prefix='/auth')
|
app.register_blueprint(auth_bp, url_prefix='/auth')
|
||||||
@@ -13,7 +12,7 @@ app.register_blueprint(auth_bp, url_prefix='/auth')
|
|||||||
@app.route('/')
|
@app.route('/')
|
||||||
@check_auth
|
@check_auth
|
||||||
def index():
|
def index():
|
||||||
return "YATTA"
|
return render_template('index.html')
|
||||||
|
|
||||||
@app.route('/logout')
|
@app.route('/logout')
|
||||||
@check_auth
|
@check_auth
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
from flask import render_template, redirect, url_for, session, request
|
from flask import redirect, url_for, session
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user