From e96605d8318493a6a18604515aaf6a411d2cde9e Mon Sep 17 00:00:00 2001
From: candifloss <candifloss@candifloss.cc>
Date: Wed, 5 Feb 2025 13:56:37 +0530
Subject: [PATCH] Check csv for existing items

---
 routes/upload.py           | 16 +++++++++++++---
 templates/csv_preview.html | 35 ++++++++++++++++++++++++++++++++++-
 2 files changed, 47 insertions(+), 4 deletions(-)

diff --git a/routes/upload.py b/routes/upload.py
index 6b98d1b..83d37b3 100644
--- a/routes/upload.py
+++ b/routes/upload.py
@@ -22,11 +22,21 @@ def upload_file():
 
         try:
             # Extract data from the CSV file
-            assets = get_csv_data(file)
-            session['assets'] = assets  # Store assets in session
+            csvdata = get_csv_data(file)
+            new_assets = []
+            existing_assets = []
+            for row in csvdata:
+                tag = row['assettag']
+                asset_exists = Asset.query.filter_by(assettag=tag).first()
+                if asset_exists:
+                    existing_assets.append(row)
+                else:
+                    new_assets.append(row)
+
+            session['assets'] = new_assets  # Store assets in session
 
             # Redirect to preview page with the CSV data
-            return render_template('csv_preview.html', assets=assets)
+            return render_template('csv_preview.html', new_assets=new_assets, existing=existing_assets)
 
         except Exception as e:
             # Handle errors during file processing
diff --git a/templates/csv_preview.html b/templates/csv_preview.html
index f56dbfd..ee7aa02 100644
--- a/templates/csv_preview.html
+++ b/templates/csv_preview.html
@@ -10,6 +10,8 @@
     <h1>CSV Preview</h1>
 
     <!-- Display CSV data in a table -->
+    {% if new_assets %}
+    <p>New assets:</p>
     <table border="1">
         <thead>
             <tr>
@@ -21,7 +23,7 @@
             </tr>
         </thead>
         <tbody>
-            {% for asset in assets %}
+            {% for asset in new_assets %}
                 <tr>
                     <td contenteditable="true">{{ asset.assettag }}</td>
                     <td contenteditable="true">{{ asset.hostname }}</td>
@@ -32,10 +34,41 @@
             {% endfor %}
         </tbody>
     </table>
+    {% endif %}
+
+    {% if existing %}
+    <p>These assets are already in the database:</p>
+    <table border="1">
+        <thead>
+            <tr>
+                <th>Asset Tag</th>
+                <th>Hostname</th>
+                <th>Warranty From</th>
+                <th>Status</th>
+                <th>Staff Number</th>
+            </tr>
+        </thead>
+        <tbody>
+            {% for asset in existing %}
+                <tr>
+                    <td>{{ asset.assettag }}</td>
+                    <td>{{ asset.hostname }}</td>
+                    <td>{{ asset.warrantyfrom }}</td>
+                    <td>{{ asset.status }}</td>
+                    <td>{{ asset.staffnum }}</td>
+                </tr>
+            {% endfor %}
+        </tbody>
+    </table>
+    {% endif %}
 
     <!-- Form button to confirm and save data -->
+    {% if new_assets %}
     <form action="/confirm_save" method="POST" onsubmit="collectEditedData()">
         <button type="submit">Confirm and Save to Database</button>
     </form>
+    {% endif %}
+    <!-- Cancel button that redirects to the home page -->
+    <button type="button" onclick="window.location.href='/'">Cancel</button>
 </body>
 </html>
\ No newline at end of file