Add files via upload

Adding basing frontend logic based on flask / jinja2.

1 files changed, 189 insertions, 0 deletions

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..8969b5d
--- /dev/null
+++ b/main.py
@@ -0,0 +1,189 @@
+# test.py
+from app import app
+from db_setup import init_db, db_session
+from forms import JusticeSearchForm, CompanyForm
+from flask import flash, render_template, request, redirect
+# from models import Company, Soud
+from models import Company, Obce, Ulice, Pravni_Forma, Insolvency_Events
+from tables import Results
+
+init_db()
+
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    search = JusticeSearchForm(request.form)
+    print(search)
+    if request.method == 'POST':
+        return search_results(search)
+
+    return render_template('index.html', form=search)
+
+@app.route('/results')
+def search_results(search):
+    results = []
+    ico = search.ico_search.data
+    ico_search_method = search.ico_search_selection.data
+    nazev = search.nazev_subjektu.data
+    nazev_search_method = search.nazev_subjektu_selection.data
+    oddil = search.oddil_search.data
+    oddil_search_method = search.oddil_search_selection.data
+    vlozka = search.vlozka_search.data
+    vlozka_search_method = search.vlozka_search_selection.data
+    obec = search.obec_search.data
+    obec_search_method = search.obec_search_selection.data
+    ulice = search.ulice_search.data
+    ulice_search_method = search.ulice_search_selection.data 
+    pravni_forma = search.pravni_forma_search.data 
+    soud = search.soud_search.data
+    insolvent_only = search.insolvent_only_search.data
+    if insolvent_only == False:
+        qry = Company.query.join(Obce, Company.obec).join(Ulice, Company.ulice).join(Pravni_Forma, Company.pravni_forma).join(Insolvency_Events, isouter=True)
+    else:
+        qry = Company.query.join(Obce, Company.obec).join(Ulice, Company.ulice).join(Pravni_Forma, Company.pravni_forma).join(Insolvency_Events)
+    if ico:
+        if ico_search_method == "text_anywhere":
+            qry = qry.filter(Company.ico.contains(ico))
+        elif ico_search_method == "text_beginning":
+            qry = qry.filter(Company.ico.like(f'{ico}%'))
+        elif ico_search_method == "text_exact":
+            qry = qry.filter(Company.ico == ico)
+    if nazev:
+        if nazev_search_method == "text_anywhere":
+            qry = qry.filter(Company.nazev.contains(nazev))
+        elif nazev_search_method == "text_beginning":
+            qry = qry.filter(Company.nazev.like(f'{nazev}%'))
+        elif nazev_search_method == "text_exact":
+            qry = qry.filter(Company.nazev == nazev)
+    if oddil:
+        if oddil_search_method == "text_anywhere":
+            qry = qry.filter(Company.oddil.contains(oddil))
+        elif oddil_search_method == "text_beginning":
+            qry = qry.filter(Company.oddil.like(f'{oddil}%'))
+        elif oddil_search_method == "text_exact":
+            qry = qry.filter(Company.oddil == oddil)                        
+        # qry = qry.filter(Company.oddil.contains(oddil))
+    if vlozka:
+        if vlozka_search_method == "text_anywhere":
+            qry = qry.filter(Company.vlozka.contains(vlozka))
+        elif vlozka_search_method == "text_beginning":
+            qry = qry.filter(Company.vlozka.like(f'{vlozka}%'))
+        elif vlozka_search_method == "text_exact":
+            qry = qry.filter(Company.vlozka == vlozka)           
+        
+        # qry = qry.filter(Company.vlozka.contains(vlozka))
+    if obec:
+        if obec_search_method == "text_anywhere":
+            qry = qry.filter(Obce.obec_jmeno.contains(obec))
+        elif obec_search_method == "text_beginning":
+            qry = qry.filter(Obce.obec_jmeno.like(f'{obec}%'))
+        elif obec_search_method == "text_exact":
+            qry = qry.filter(Obce.obec_jmeno == obec)              
+        # qry = qry.filter(Obce.obec_jmeno.contains(obec))
+    if ulice:
+        if ulice_search_method == "text_anywhere":
+            qry = qry.filter(Ulice.ulice_jmeno.contains(ulice))
+        elif ulice_search_method == "text_beginning":
+            qry = qry.filter(Ulice.ulice_jmeno.like(f'{ulice}%'))
+        elif ulice_search_method == "text_exact":
+            qry = qry.filter(Ulice.ulice_jmeno == ulice)               
+        
+        # qry = qry.filter(Ulice.ulice_jmeno.contains(ulice))
+    if pravni_forma:
+        qry = qry.filter(Pravni_Forma.pravni_forma.contains(pravni_forma))
+    if soud:
+        qry = qry.filter(Company.soud.contains(soud))
+    results = qry.all()
+        # else:
+        #     qry = db_session.query(Company)
+        #     results = qry.all()
+    # else:
+    #     qry = db_session.query(Company)
+    #     results = qry.all()
+    
+    if not results:
+        flash('No results found!')
+        return redirect('/')
+    
+    else:
+        table = Results(results)
+        table.border = True
+        # return render_template('results.html', table=table)
+        return render_template("results2.html", results=results, form=search)
+
+def search_results_BACKUP(search):
+    results = []
+    search_string = search.data['search']
+    
+    if search_string:
+        # if search.data['select'] == 'soud':
+        #     qry = db_session.query(Company, Soud).filter(
+        #         Soud.id==Company.soud_id).filter(
+        #             Soud.name.contains(search_string))
+        #     results = [item[0] for item in qry.all()]
+        if search.data['select'] == 'nazev':
+            qry = db_session.query(Company).filter(
+                Company.nazev.contains(search_string))
+            results = qry.all()
+        elif search.data['select'] == 'ico':
+            qry = db_session.query(Company).filter(
+                Company.ico.contains(search_string))
+            results = qry.all()
+        else:
+            qry = db_session.query(Company)
+            results = qry.all()
+    else:
+        qry = db_session.query(Company)
+        results = qry.all()
+    
+    if not results:
+        flash('No results found!')
+        return redirect('/')
+    
+    else:
+        table = Results(results)
+        table.border = True
+        return render_template('results.html', table=table)
+
+
+@app.route('/new_company', methods=['GET', 'POST'])
+def new_company():
+    """
+    Add a new company
+    """
+    form = CompanyForm(request.form)
+    
+    if request.method == 'POST' and form.validate():
+        # save the album
+        company = Company()
+        save_changes(company, form, new=True)
+        flash('Company created successfully!')
+        return redirect('/')
+    return render_template('new_company.html', form=form)
+
+def save_changes(company, form, new=False):
+    """
+    Save the changes to the database
+    """
+    # Get data from form and assign it to the correct attributes
+    # of the SQLAlchemy table object
+    # soud = Soud()
+    # soud.name = form.soud.data
+    
+    # company.soud = soud
+    company.soud = form.soud.data
+    company.ico = form.ico.data
+    company.nazev = form.nazev.data
+    company.sidlo = form.sidlo.data
+    company.zapis = form.zapis.data
+    company.oddil = form.oddil.data
+    company.vlozka = form.vlozka.data
+    company.vymaz = form.vymaz.data
+    if new:
+        # Add the new album to the database
+        db_session.add(company)
+    # commit the data to the database
+    db_session.commit()
+
+
+if __name__ == '__main__':
+    app.run()
\ No newline at end of file