Added check for empty db

main.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, send_from_directory, request
+from flask import Flask, redirect,url_for,  render_template, send_from_directory, request
 from flask_httpauth import HTTPBasicAuth
 from werkzeug.security import check_password_hash
 from gevent.pywsgi import WSGIServer
@@ -13,6 +13,7 @@ import re
 import datetime
 import sys
 import time
+import numpy as np
 from pathlib import Path
 from io import BytesIO
 
@@ -35,12 +36,38 @@ def verify_password(username, password):
     ):
         return username
 
-
+@app.route("/", methods=['POST','GET'])
-@app.route("/")
 def startpage():
     #result = "Hello, World!"
+    config._print(request.method)
+    if request.method == 'POST':
+        if request.form.get('Create') == 'Create':
+            # pass
+            config._print("open")
             conn = sqlite3.connect('app.db')
             cursor = conn.cursor()
+        
+            cursor.execute("create table COMICS (CVDB,ISSUE,SERIES,VOLUME, PUBLISHER, TITLE, FILE,PATH,UPDATED,PRIMARY KEY(CVDB))")
+            result = cursor.fetchall()
+            conn.close()
+            config._print("Encrypted")
+        elif  request.form.get('Import') == 'Import':
+            # pass # do something else
+            config._print("Decrypted")
+            return redirect(url_for('import2sql'))
+        elif request.form.get('Generate') == 'Generate':
+            config._print("Generate Covers from Start page")
+            return redirect(url_for('generate'))
+        else:
+            # pass # unknown
+            return render_template("first.html")
+    elif request.method == 'GET':
+        # return render_template("index.html")
+        config._print("No Post Back Call")
+    conn = sqlite3.connect('app.db')
+    cursor = conn.cursor()
+    
+    try:
         cursor.execute("select * from comics LIMIT " + str(config.DEFAULT_SEARCH_NUMBER) + ";")
         result = cursor.fetchall()
     
@@ -53,13 +80,38 @@ def startpage():
         cursor.execute("SELECT volume, COUNT(volume) FROM comics GROUP BY volume ORDER BY volume;")
         volume = cursor.fetchall()
         
+        
+
         x = []
         y = []
         for i in volume:
             x.append(i[0])
             y.append(i[1])
         conn.close()
-    return render_template("start.html", result=result,pub_list=pub_list,count=count,x=x,y=y)
+
+        try:
+            total = np.sum(np.array(volume).astype('int')[:,1],axis=0)
+            dir_path = r'thumbnails'
+            covers = 0
+            for path in os.listdir(dir_path):
+                if os.path.isfile(os.path.join(dir_path,path)):
+                    covers += 1
+            config._print("covers: " + str(covers))
+        except Exception as e:
+            config._print(e)
+        return render_template("start.html", first=False,result=result,pub_list=pub_list,count=count,x=x,y=y,total=total,covers=covers)
+    except:
+        conn.close()
+
+        config._print('first')
+        return render_template("start.html",first=True)
+
+#@app.route("/first", methods=['GET', 'POST'])
+#def first():
+#   return render_template('first.html',result=result) 
+
+
+
 
 @app.route("/healthz")
 def healthz():
@@ -85,7 +137,8 @@ def generate():
                         Bs_data = BeautifulSoup(s.open('ComicInfo.xml').read(), "xml")
                         CVDB=extras.get_cvdb(Bs_data.select('Notes'))
                         if force == 'True':
-                            cover = s.open(filelist[1]).read()
+                            ext = [i for i, x in enumerate(filelist) if re.search("(?i)\.jpg|png|jpeg$", x)]
+                            cover = s.open(filelist[ext[0]]).read()
 
                             image = Image.open(BytesIO(cover))
                             image.thumbnail(config.MAXSIZE,Image.ANTIALIAS)
@@ -97,7 +150,13 @@ def generate():
                             #c.close()
                             generated = generated + 1
                         elif Path(config.THUMBNAIL_DIR + "/" + str(CVDB) + ".jpg").exists() == False:
-                            cover = s.open(filelist[1]).read()
+                            ext = [i for i, x in enumerate(filelist) if re.search("(?i)\.jpg|png|jpeg$", x)]
+                            config._print(filelist)
+                            config._print(ext)
+                            config._print(filelist[ext[0]])
+                            cover = s.open(filelist[ext[0]]).read()
+                            #xyz = [i for i, x in enumerate(filelist) if re.match('*\.py$',x)]
+                            #config._print(xyz)
                             image = Image.open(BytesIO(cover))
                             image.thumbnail(config.MAXSIZE,Image.ANTIALIAS)
                             image.save(config.THUMBNAIL_DIR + "/" + str(CVDB) + ".jpg")
@@ -198,16 +257,16 @@ def image(path):
 @app.route("/catalog/<path:path>")
 @auth.login_required
 def catalog(path=""):
-    config._print("path: " + path)
+    #config._print("path: " + path)
-    config._print("root_url: " + request.root_url)
+    #config._print("root_url: " + request.root_url)
-    config._print("url: " + request.url)
+    #config._print("url: " + request.url)
-    config._print("CONTENT_BASE_DIR: " + config.CONTENT_BASE_DIR)
+    #config._print("CONTENT_BASE_DIR: " + config.CONTENT_BASE_DIR)
     #print("PRESSED ON")
     start_time = timeit.default_timer()
     #print(request.root_url) 
     c = fromdir(request.root_url, request.url, config.CONTENT_BASE_DIR, path)
-    print("c: ")
+    #print("c: ")
-    pprint(vars(c))
+    #pprint(vars(c))
     for x in c.entries:
         for y in x.links:
             pprint(y.href)

templates/first.html

@@ -0,0 +1,14 @@
+<html>
+    <body>
+
+        <form method="post" action="/">
+
+                <input type="submit" value="Encrypt" name="Encrypt"/>
+                <input type="submit" value="Decrypt" name="Decrypt" />
+
+        </form>
+    </body>
+</html>
+    
+
+<p>{{ result }}</p>

templates/start.html

@@ -2,6 +2,27 @@
 <script src="https://cdnjs.cloudflare.com/ajax/libs/Chart.js/2.5.0/Chart.min.js"></script>
 <body>
 
+  {% if first and request.args.get('first') == None %}
+    <form method="post">
+        <p>DB is missing table. <input type="submit" value="Create" name="Create"/>
+    </form>
+    {% endif %}
+
+  {% if result == [] %}
+
+    <form method="post">
+        <p>No comics imported. <input type="submit" value="Import" name="Import"/>
+    </form>
+    {% endif %}
+
+    {% if total != covers %}
+    <form method="post">
+      <p>Some covers missing <input type="submit" value="Generate" name="Generate"/>
+    </form>
+  {% endif %}
+
+    <h1>Total Comics: {{ total }}</h1>
+
 <canvas id="myChart" style="width:100%;max-width:600px"></canvas>
 
 <script>