fix: base64 encoding at update pray

PrayHelper · Nov 8, 2023 · 89c04d5 · 89c04d5
1 parent b022f99
commit 89c04d5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/app/utils/pray.py b/app/utils/pray.py
@@ -265,7 +265,8 @@ def update_storage(storage_id, content) -> StorageDTO:
             if 'pray_cnt' in content:
                 storage.pray_cnt = content['pray_cnt']
             db.session.commit()
-        except Exception:
+        except Exception as E:
+            print(E)
             raise StorageFail('update storage error')
         return StorageDTO(
                 id=storage.id,
@@ -382,14 +383,15 @@ def update_pray(content, storage_id) -> PrayDTO:
                 raise PrayFail('can not update other user pray')
             pray = Pray.query.filter_by(id=storage.pray_id).first()
             if 'target' in content:
-                pray.target = content['target']
+                pray.target =  base64.b64encode(content['target'].encode('utf-8')).decode('utf-8')
             if 'title' in content:
-                pray.title = content['title']
+                pray.title = base64.b64encode(content['title'].encode('utf-8')).decode('utf-8')
             if 'deadline' in content:
                 storage.deadline = content['deadline']
             db.session.commit()
             return StorageService.get_storage(storage_id)
 
-        except Exception:
+        except Exception as E:
+            print(E)
             raise PrayFail('update pray error')