diff --git a/app.py b/app.py
index f1f1c5c..319db75 100644
--- a/app.py
+++ b/app.py
@@ -3,25 +3,45 @@ import json
app = Flask(__name__)
+tmp = '71386-10'
+
@app.route('/')
def index():
- return render_template('index.html')
+ with open('./sets/'+tmp+'/info.json') as info:
+ info_file = json.loads(info.read())
+ with open('./sets/'+tmp+'/inventory.json') as inventory:
+ inventory_file = json.loads(inventory.read())
+ return render_template('bootstrap_table.html', title=info_file['set_num']+" - "+info_file['name'],
+ info_file=info_file,inventory_file=inventory_file)
+
+ #return render_template('index.html')
#'Welcome to the Flask App'
@app.route('/saveNumber', methods=['POST'])
def save_number():
- data = request.get_json()
+ data1 = request.form.get('brick.part.part_num')
+ data2 = request.form.get('brick.color.name')
number = request.form.get('numberInput')
if number is not None:
+
+ print(data1)
+ print(data2)
+ print(number)
+
+ with open('./info/'+tmp+'.json') as info:
+ json_file = json.loads(info.read())
+ print(json_file['count'])
+
+ print(json_file['unit'][0]['bricks']['missing'])
+
# Save number to JSON file
with open('data.json', 'w') as json_file:
json.dump({'number': number}, json_file)
- return jsonify({'message': 'Number saved successfully'}), 200
- else:
- return jsonify({'error': 'Invalid data provided'}), 400
+
+ return ('', 204)
if __name__ == '__main__':
- app.run(debug=True)
+ app.run(host='192.168.10.109', debug=True, port=3333)
diff --git a/lego.py b/lego.py
index 45d88bc..a8ea4e7 100644
--- a/lego.py
+++ b/lego.py
@@ -7,8 +7,6 @@ import rebrick #rebrickable api
# json things
import json
-from bs4 import BeautifulSoup
-
import requests # request img from web
import shutil # save img locally
@@ -34,9 +32,11 @@ with open('api','r') as f:
rb = rebrick.init(api_key)
Path(set_path).mkdir(parents=True, exist_ok=True)
+Path('./static/parts').mkdir(parents=True, exist_ok=True)
+Path('./info').mkdir(parents=True, exist_ok=True)
# Get set info
-response = json.loads(rebrick.lego.get_set(int(set_num)).read())
+response = json.loads(rebrick.lego.get_set(set_num).read())
with open(set_path+'info.json', 'w', encoding='utf-8') as f:
json.dump(response, f, ensure_ascii=False, indent=4)
@@ -53,17 +53,17 @@ else:
logging.error('set_img_url: ' + set_num)
# set inventory
-response = json.loads(rebrick.lego.get_set_elements(int(set_num)).read())
+response = json.loads(rebrick.lego.get_set_elements(set_num).read())
with open(set_path+'inventory.json', 'w', encoding='utf-8') as f:
json.dump(response, f, ensure_ascii=False, indent=4)
# get part images if not exists
for i in response["results"]:
- if not Path("./parts/"+i["element_id"]+".jpg").is_file():
+ if not Path("./static/parts/"+i["element_id"]+".jpg").is_file():
res = requests.get(i["part"]["part_img_url"], stream = True)
if res.status_code == 200:
- with open("./parts/"+i["element_id"]+".jpg",'wb') as f:
+ with open("./static/parts/"+i["element_id"]+".jpg",'wb') as f:
shutil.copyfileobj(res.raw, f)
print('image saved')
else:
diff --git a/templates/base.html b/templates/base.html
new file mode 100644
index 0000000..a6eea5e
--- /dev/null
+++ b/templates/base.html
@@ -0,0 +1,15 @@
+
+
+
+ {{ title }}
+
+
+
+
+
{{ title }}
+
+ {% block content %}{% endblock %}
+
+ {% block scripts %}{% endblock %}
+
+
diff --git a/templates/bootstrap_table.html b/templates/bootstrap_table.html
new file mode 100644
index 0000000..a7de2bd
--- /dev/null
+++ b/templates/bootstrap_table.html
@@ -0,0 +1,42 @@
+{% extends "base.html" %}
+
+{% block content %}
+
+
+
+ | img |
+ part.element_id |
+ part.part_num |
+ part.name |
+ color.id |
+ color.name |
+ part.quantity |
+ part.is_spare |
+ Missing? |
+
+
+
+ {% for brick in inventory_file.results %}
+
+  |
+ {{ brick.element_id }} |
+ {{ brick.part.part_num }} |
+ {{ brick.part.name }} |
+ {{ brick.color.id }} |
+ {{ brick.color.name }} |
+ {{ brick.quantity }} |
+ {{ brick.is_spare }} |
+
+
+ |
+
+ {% endfor %}
+
+
+{% endblock %}
\ No newline at end of file