Fixed missing part image and added conditions for missing part url

app.py

@@ -78,8 +78,11 @@ def create():
             set_img_url,
             set_url,
             last_modified_dt,
+            mini_col,
+            set_check,
+            set_col,
             u_id
-            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) ''', (response['set_num'], response['name'], response['year'], response['theme_id'], response['num_parts'],response['set_img_url'],response['set_url'],response['last_modified_dt'],unique_set_id))
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''', (response['set_num'], response['name'], response['year'], response['theme_id'], response['num_parts'],response['set_img_url'],response['set_url'],response['last_modified_dt'],False,False,False,unique_set_id))
 
         conn.commit()
 
@@ -108,16 +111,27 @@ def create():
         for i in response['results']:
             # Get part image. Saved under ./static/parts/xxxx.jpg
             part_img_url = i['part']['part_img_url']
+            part_img_url_id = 'nil'
 
-            pattern = r'/([^/]+)\.(?:png|jpg)$'
+            try:
-            match = re.search(pattern, part_img_url)
+                pattern = r'/([^/]+)\.(?:png|jpg)$'
+                match = re.search(pattern, part_img_url)
 
-            if match:
+                if match:
-                part_img_url_id = match.group(1)
+                    part_img_url_id = match.group(1)
-                print("Part number:", part_img_url_id)
+                    print("Part number:", part_img_url_id)
-            else:
+                else:        
-                print("Part number not found in the URL.")
+                    print("Part number not found in the URL.")
-                print(">>> " + part_img_url)
+                    print(">>> " + part_img_url)
+            except Exception as e:
+                    print("Part number not found in the URL.")
+                    print(">>> " + str(part_img_url))
+                    print(str(e))
+
+
+            
+
+            
 
 
             cursor.execute('''INSERT INTO inventory (
@@ -138,17 +152,20 @@ def create():
             
             if not Path("./static/parts/"+part_img_url_id+".jpg").is_file():
                 print('Saving part image:',end='')
-
+                if part_img_url is not None:
-                res = requests.get(part_img_url, stream = True)
+                    res = requests.get(part_img_url, stream = True)
-                count+=1
+                    count+=1
-                if res.status_code == 200:
+                    if res.status_code == 200:
-                    with open("./static/parts/"+part_img_url_id+".jpg",'wb') as f:
+                        with open("./static/parts/"+part_img_url_id+".jpg",'wb') as f:
-                        shutil.copyfileobj(res.raw, f)
+                            shutil.copyfileobj(res.raw, f)
-                        print(' OK')
+                            print(' OK')
+                    else:
+                        print('Image Couldn\'t be retrieved for set ' + part_img_url_id)
+                        logging.error('part_img_url: ' + part_img_url_id)
+                        print(' ERROR')
                 else:
-                    print('Image Couldn\'t be retrieved for set ' + part_img_url_id)
+                    print('Part url is None')
-                    logging.error('part_img_url: ' + part_img_url_id)
+                    print(i)
-                    print(' ERROR')
             else: 
                 print(part_img_url_id + '.jpg exists!')
 

templates/index.html

@@ -204,7 +204,7 @@
 	      <div class="columns" style="">
 		<div class="column is-half" style="">
 		  <figure class="image is-4by3">
-			  <a href="/{{ i[0] }}/{{ i[8] }}">
+			  <a href="/{{ i[0] }}/{{ i[12] }}">
 		      <img style='height: 100%; width: 100%; object-fit: contain'  src="/static/sets/{{ i[0] }}.jpg" alt="Image">
 		    </a>
 		  </figure>
@@ -239,7 +239,7 @@
 		<footer class="card-footer" style="">
 		  <p class="card-footer-item">
 		  <span>
-			<span class="is-size-7">{{ i[8] }}</span>
+			<span class="is-size-7">{{ i[12] }}</span>
 			  
 		  </span>
 		  </p>

templates/table.html

@@ -16,7 +16,13 @@
     <tbody>
       {% for brick in inventory_list %}
         <tr>
+
+          {% if brick[5] == 'nil' %}
+          <td><img src="{{ '/static/none.jpg' }}" class="lightbox-trigger" alt="{{ brick[3] }}" style="height: 50px; width: 50px;margin:0;padding: 0;"></td>
+          {% else %}
           <td><img src="{{ '/static/parts/' + brick[5] + '.jpg' }}" class="lightbox-trigger" alt="{{ brick[3] }}" style="height: 50px; width: 50px;margin:0;padding: 0;"></td>
+          {% endif %}
+
           <td style="text-align:left;margin:0px;" class="hidden-mobile name-class">{{ brick[3] }}</td>
           <td class="hidden-mobile">{{ brick[7] }}</td>
           <td style="text-align: center;">{{ brick[8] }}</td>