浏览代码

wrap account assets response in JSON object

kunickyd 3 年之前
父节点
当前提交
bda961581e
共有 1 个文件被更改,包括 7 次插入3 次删除
  1. 7 3
      app.py

+ 7 - 3
app.py

@@ -37,6 +37,9 @@ class Asset:
         self.assetId = assetId
         self.balance = balance
 
+class Assets:
+    assets = []
+
 @app.get("/accounts/<string:accountId>/assets/")
 def get_account_assets(accountId): #TODO: add validation and error handling
 
@@ -45,10 +48,11 @@ def get_account_assets(accountId): #TODO: add validation and error handling
 
     iroha_response = net.send_query(query)
     data = iroha_response.account_assets_response.account_assets
-
-    response = []
+    
+    response = Assets()
+    response.assets = []
 
     for asset in data:
-        response.append(Asset(asset.asset_id, asset.balance))
+        response.assets.append(Asset(asset.asset_id, asset.balance))
 
     return jsonify(response)