mirror of
https://github.com/Llloooggg/Jarvis.git
synced 2026-03-06 03:56:23 +03:00
Дописал сор данных с полей
This commit is contained in:
29
__init__.py
29
__init__.py
@@ -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)
|
||||||
|
|||||||
@@ -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()
|
||||||
|
|||||||
@@ -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">
|
||||||
|
|||||||
Reference in New Issue
Block a user