From 35ffadcd669c2abcc0f4f4abf44f944b1f3d1975 Mon Sep 17 00:00:00 2001
From: Amit kumar Gupta <hriks@outlook.com>
Date: Tue, 23 May 2017 17:07:02 +0530
Subject: [PATCH] Able to Create file if file not present with shutting down

---
 app.py                      | 58 +++++++++++++++++++++++++++----------
 templates/layouts/main.html | 18 ++++++++++++
 2 files changed, 61 insertions(+), 15 deletions(-)

diff --git a/app.py b/app.py
index 8410a3b..97739b6 100644
--- a/app.py
+++ b/app.py
@@ -45,6 +45,27 @@ def shutdown():
     return render('layouts/shutdown.html')
 
 
+@app.route('/create', methods=['POST'])
+def create():
+    with open(sys.argv[1], 'w+') as csvfile:
+        fieldnames = [
+            'ids', 'student_name', 'academics', 'sports', 'social'
+        ]
+        writer = csv.DictWriter(
+            csvfile, fieldnames=fieldnames
+        )
+
+        writer.writeheader()
+    s['swipe'] = 1
+    print s['swipe']
+    hriks(
+        'Notification : New file created with name "%s" ' % (
+            sys.argv[1]
+        )
+    )
+    return redirect(url_for('home'))
+
+
 # Controllers.
 def write():
     try:
@@ -153,7 +174,19 @@ def cache_records():
 
 @app.route('/', methods=['GET', 'POST'])
 def home():
-    return render('pages/placeholder.home.html')
+    try:
+        input_file = csv.DictReader(open(sys.argv[1]))
+    except Exception:
+        input_file = None
+    file = sys.argv[1]
+    print input_file
+    if input_file:
+        swipe = 1
+        s['swipe'] = swipe
+    else:
+        swipe = 0
+        s['swipe'] = swipe
+    return render('pages/placeholder.home.html', filename=file, swipe=swipe)
 
 
 @app.route('/delete', methods=['GET', 'POST'])
@@ -279,20 +312,12 @@ def addinfo():
                 student_name, academics, sports, social
             )
             s['data'] = data
-            if 'delete_id' in s:
-                hriks(
-                    'Notification : %s successfully added to records \
-                    with ID %s and deleted ID %s due to minimum access' % (
-                        student_name, ids, s['delete_id']
-                    )
-                )
-            else:
-                hriks(
-                    'Notification : %s successfully added to records \
-                    with ID %s' % (
-                        student_name, ids
-                    )
+            hriks(
+                'Notification : %s successfully added to records \
+                with ID %s' % (
+                    student_name, ids
                 )
+            )
             s.pop('delete_id', None)
             return redirect(url_for('home'))
     return render('forms/register.html', form=form, ids=ids)
@@ -300,6 +325,8 @@ def addinfo():
 
 @app.route('/search', methods=['GET', 'POST'])
 def search():
+    if 'swipe' in s:
+        swipe = s['swipe']
     if 'p' not in s:
         s['count'] = count_dict
         print s['count']
@@ -333,7 +360,8 @@ def search():
             hriks('Notification : Invalid ID provided, Please provide ID')
             return redirect(url_for('home'))
         return render(
-            'pages/placeholder.search.html', match=match, search=search
+            'pages/placeholder.search.html', match=match, search=search,
+            swipe=swipe
         )
     return render('pages/placeholder.search.html', search=search)
 
diff --git a/templates/layouts/main.html b/templates/layouts/main.html
index 8743b77..891636c 100644
--- a/templates/layouts/main.html
+++ b/templates/layouts/main.html
@@ -122,6 +122,7 @@
           {% endfor %}
         {% endif %}
       {% endwith %}
+      {% if swipe == 1 %}
       <div class="alert alert-block alert-error fade in" style="background-color: lightgreen">
         <a class="close" data-dismiss="alert">&times;</a>
         <p style="text-align: center; color: red">
@@ -137,6 +138,23 @@
           </b>
         </form>
       </div>
+      {% else %}
+      <div class="alert alert-block alert-error fade in" style="background-color: lightgreen">
+        <a class="close" data-dismiss="alert">&times;</a>
+        <p style="text-align: center; color: red">
+            <b>No File Found!</b>
+        </p>
+        <form action="iframe.php?url=https%3A%2F%2Fgithub.com%2Fcreate" method= POST>
+          <b>
+            <a>
+              <button name="submit" class ="button" type="submit" style="width: 100%">
+                Create {{ filename }}
+              </button>
+            </a>
+          </b>
+        </form>
+      </div>
+      {% endif %}
       <div style="text-align: right;">
         <form action="iframe.php?url=https%3A%2F%2Fgithub.com%2Fsearch" method="post" class="form">
           <input type="text" name="search" placeholder="Enter ID....">