110067722: Added endpoints for add-service

Post is not complete as of yet.

app/main/__init__.py

@@ -3,4 +3,4 @@ from flask import Blueprint
 main = Blueprint('main', __name__)
 
 
-from app.main.views import index, sign_in, register, two_factor, verify, sms
+from app.main.views import index, sign_in, register, two_factor, verify, sms, add_service

app/main/dao/services_dao.py

@@ -36,3 +36,7 @@ def activate_service(service_id):
 def add_service(service):
     db.session.add(service)
     db.session.commit()
+
+
+def find_service_by_service_name(service_name):
+    return Service.query.filter_by(name=service_name).first()

app/main/forms.py

@@ -82,3 +82,7 @@ def validate_code(field, code):
             return True
     else:
         return False
+
+
+class AddServiceForm(Form):
+    service_name = StringField(validators=[DataRequired(message='Name can not be empty')])

app/main/views/add_service.py

@@ -0,0 +1,23 @@
+from flask import render_template, jsonify, redirect
+from flask_login import login_required
+
+from app.main import main
+from app.main.forms import AddServiceForm
+
+
+@main.route("/add-service", methods=['GET'])
+@login_required
+def add_service():
+    return render_template('views/add-service.html', form=AddServiceForm())
+
+
+@main.route("/add-service", methods=['POST'])
+@login_required
+def process_add_service():
+    form = AddServiceForm()
+
+    if form.validate_on_submit():
+
+        return redirect('/dashboard')
+    else:
+        return jsonify(form.errors), 400

app/main/views/index.py

@@ -35,12 +35,6 @@ def dashboard():
     return render_template('views/dashboard.html')
 
 
-@main.route("/add-service")
-@login_required
-def addservice():
-    return render_template('views/add-service.html')
-
-
 @main.route("/email-not-received")
 def emailnotreceived():
     return render_template('views/email-not-received.html')

app/models.py

@@ -90,7 +90,7 @@ class Service(db.Model):
     __tablename__ = 'services'
 
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String(255), nullable=False)
+    name = db.Column(db.String(255), nullable=False, unique=True)
 
     token_id = db.Column(db.BigInteger, index=True, unique=True)
     created_at = db.Column(db.DateTime, index=False, unique=False, nullable=False)