From aa9212f3226b10976c31cdf804139d60ae655a16 Mon Sep 17 00:00:00 2001
From: Andrew <saintruler@gmail.com>
Date: Sun, 21 Jul 2019 13:14:38 +0400
Subject: =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5=D0=BD=D0=B0?=
 =?UTF-8?q?=20=D0=B2=D0=B0=D0=BB=D0=B8=D0=B4=D0=B0=D1=86=D0=B8=D1=8F=20?=
 =?UTF-8?q?=D0=BF=D1=80=D0=B8=20=D0=BE=D0=B1=D0=BD=D0=BE=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=B8=D0=B8=20=D0=B7=D0=BD=D0=B0=D1=87=D0=B5=D0=BD=D0=B8?=
 =?UTF-8?q?=D0=B9=20=D0=B2=20=D1=80=D1=8F=D0=B4=D1=83.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 day9/task5_vue/backend/views.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

(limited to 'day9/task5_vue/backend/views.py')

diff --git a/day9/task5_vue/backend/views.py b/day9/task5_vue/backend/views.py
index d6d9351..ab8b628 100644
--- a/day9/task5_vue/backend/views.py
+++ b/day9/task5_vue/backend/views.py
@@ -67,16 +67,22 @@ def validate(query, *args):
 
 def update_post(query, *args):
     wrapper = get_wrapper_for('mysql')
-    service_id = query['service_id']
+    scheme = wrapper.schemes['table_task1']
+
     permitted_fields = wrapper.get_column_names()
     permitted_fields.remove('service_id')
 
-    query_set = []
-    for field_name, value in query.items():
-        if field_name in permitted_fields:
-            query_set.append(f'{field_name}="{value}"')
+    validation_results = scheme.validate(query)
+    if not validation_results['error']:
+        expressions = {}
+        for key, value in query.items():
+            if key in permitted_fields:
+                expressions[key] = value
+
+        wrapper.update('table_task1', expressions, {'service_id': query['service_id']})
+    else:
+        return ErrorResponse(500)
 
-    wrapper.update('table_task1', query_set, {'service_id': service_id})
     return HtmlResponse('<a href="/">Return to main page</a><br><h1>Database Updated</h1>')
 
 
-- 
cgit v1.2.3