Дописал сор данных с полей

This commit is contained in:
2020-03-18 20:45:33 +03:00
parent a9d5d8572b
commit c3869f16a6
3 changed files with 53 additions and 15 deletions

View File

@@ -25,9 +25,7 @@ def registration():
userName = request.form['RegUserLogin'] userName = request.form['RegUserLogin']
userPassw = request.form['RegUserPassw'] userPassw = request.form['RegUserPassw']
if string_check(userName) and string_check(userPassw): if string_check(userName) and string_check(userPassw):
print('попытка регистрации')
if db_routing.add_user(userName, passw_hash(userPassw)): if db_routing.add_user(userName, passw_hash(userPassw)):
print(db_routing.get_user(username=userName))
login_user(db_routing.get_user(username=userName)) login_user(db_routing.get_user(username=userName))
return redirect(url_for('workshop')) return redirect(url_for('workshop'))
return render_template('registration.html') return render_template('registration.html')
@@ -58,20 +56,35 @@ def logout():
@app.route('/workshop', methods=['GET', 'POST']) @app.route('/workshop', methods=['GET', 'POST'])
@login_required @login_required
def workshop(): def workshop():
"""
if request.method == 'POST': if request.method == 'POST':
if request.form['NewSceanrio']: if 'NewScenarioName' in request.form:
newScenarioName = request.form['NewScenarioName']
triggerName = request.form['TriggerName']
triggerArgs = request.form['TriggerArgs']
actionName = request.form['ActionName']
actionArgs = request.form['ActionArgs']
db_routing.add_scenario(current_user.get_id(), newScenarioName, triggerName, triggerArgs, actionName,
actionArgs)
return redirect(url_for('workshop'))
if request.form['NewSceanrio']: # if 'NewScenarioName' in request.form:
if request.form['NewSceanrio']: if 'TGUsername' in request.form:
""" new_tg_username = request.form['TGUsername']
db_routing.tg_username_update(current_user.get_id(), new_tg_username)
return redirect(url_for('workshop'))
triggers_list = db_routing.get_trigers() triggers_list = db_routing.get_trigers()
actions_list = db_routing.get_actions() actions_list = db_routing.get_actions()
user_scripts_list = db_routing.get_user_scripts(current_user.get_id()) user_scripts_list = db_routing.get_user_scripts(current_user.get_id())
User = db_routing.get_user(id=current_user.get_id())
if User:
tg_username = User.tg_username
print(tg_username)
else:
tg_username = None
return render_template('workshop.html', triggers_list=triggers_list, actions_list=actions_list, return render_template('workshop.html', triggers_list=triggers_list, actions_list=actions_list,
user_scripts_list=user_scripts_list) user_scripts_list=user_scripts_list, tg_username=tg_username)
# @app.errorhandler(Exception) # @app.errorhandler(Exception)

View File

@@ -13,12 +13,12 @@ class User(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True, nullable=False) username = db.Column(db.String(80), unique=True, nullable=False)
password = db.Column(db.String(120), nullable=False) password = db.Column(db.String(120), nullable=False)
telegram_username = db.Column(db.String(80), unique=True) tg_username = db.Column(db.String(80), unique=True)
def __init__(self, username, password, telegram_username=None): def __init__(self, username, password, tg_username=None):
self.username = username self.username = username
self.password = password self.password = password
self.telegram_username = telegram_username self.tg_username = tg_username
def is_active(self): def is_active(self):
return True return True
@@ -48,6 +48,7 @@ class Scenario(db.Model):
__tablename__ = 'Scenarios' __tablename__ = 'Scenarios'
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
owner_id = db.Column(db.Integer, ForeignKey('Users.id')) owner_id = db.Column(db.Integer, ForeignKey('Users.id'))
scenario_name = db.Column(db.String(80))
trigger_id = db.Column(db.Integer, ForeignKey('Triggers.id')) trigger_id = db.Column(db.Integer, ForeignKey('Triggers.id'))
trigger_args = db.Column(db.String(200)) trigger_args = db.Column(db.String(200))
action_id = db.Column(db.Integer, ForeignKey('Actions.id')) action_id = db.Column(db.Integer, ForeignKey('Actions.id'))
@@ -65,6 +66,14 @@ def add_user(user_name, passw_hash):
return False return False
def add_scenario(owner_id, scenario_name, trigger_id, trigger_args, action_id, action_args):
new_scenario = Scenario(owner_id=owner_id, scenario_name=scenario_name, trigger_id=trigger_id,
trigger_args=trigger_args, action_id=action_id, action_args=action_args)
db.session.add(new_scenario)
db.session.commit()
return new_scenario
def get_user(id=None, username=None): def get_user(id=None, username=None):
if id: if id:
return User.query.filter_by(id=id).first() return User.query.filter_by(id=id).first()
@@ -85,3 +94,15 @@ def get_actions():
def get_user_scripts(current_user_id): def get_user_scripts(current_user_id):
user_scripts_list = Scenario.query.filter_by(owner_id=current_user_id).all() user_scripts_list = Scenario.query.filter_by(owner_id=current_user_id).all()
return user_scripts_list return user_scripts_list
def delete_scenario(scenario_id):
scenario = Scenario.query.filter_by(id=scenario_id).first()
db.session.delete(scenario)
db.session.commit()
def tg_username_update(current_user_id, new_tg_username):
current_user = User.query.filter_by(id=current_user_id).first()
current_user.tg_username = new_tg_username
db.session.commit()

View File

@@ -51,13 +51,13 @@
<div class="form__field-title" > <div class="form__field-title" >
Название Название
</div> </div>
<input type="text" class="form__input-text" placeholder="Введите название сбытия" name = "TitleEvent"> <input type="text" class="form__input-text" placeholder="Введите название сбытия" name = "NewScenarioName">
</label> </label>
<label class="form__log"> <label class="form__log">
<div class="form__field-title" > <div class="form__field-title" >
Условие: Условие:
</div> </div>
<select class="form__input-text" name = "Triger"> <select class="form__input-text" name = "TriggerName">
{% for trigger in triggers_list %} {% for trigger in triggers_list %}
<option>{{ trigger.name }}</option> <option>{{ trigger.name }}</option>
{% endfor %} {% endfor %}
@@ -67,7 +67,7 @@
<div class="form__field-title" > <div class="form__field-title" >
Действие: Действие:
</div> </div>
<select class="form__input-text" name = "Triger"> <select class="form__input-text" name = "ActionName">
{% for action in actions_list %} {% for action in actions_list %}
<option>{{ action.name }}</option> <option>{{ action.name }}</option>
{% endfor %} {% endfor %}
@@ -116,7 +116,11 @@
</div> </div>
<form action="/workshop" name="Telegram" method="POST"> <form action="/workshop" name="Telegram" method="POST">
<label class="form__log"> <label class="form__log">
<input type="text" class="form__input-text" placeholder="Логин" name = "TelegUserLogin"> {% if tg_username %}
<input type="text" class="form__input-text" placeholder={{tg_username}} name = "TGUsername">
{% else %}
<input type="text" class="form__input-text" placeholder="Логин" name = "TGUsername">
{% endif %}
</label> </label>
<div class="form__btns"> <div class="form__btns">
<div class="form__controls"> <div class="form__controls">