Commit to save code [2 TEST PASSES]

This commit is contained in:
andrei 2022-02-17 16:56:46 +02:00
parent e7c18545b3
commit 229872fb29
2 changed files with 6 additions and 5 deletions

View File

@ -131,9 +131,9 @@ def get_metadata(transaction_ids: list, space):
# @register_query(LocalMongoDBConnection) # @register_query(LocalMongoDBConnection)
def store_asset(asset: dict, connection): def store_asset(asset: dict, connection):
space = connection.space("assets") space = connection.space("assets")
unique = token_hex(8) # unique = token_hex(8)
try: try:
space.insert((asset["id"], unique, asset["data"])) space.insert((asset["id"], asset["data"]))
except: # TODO Add Raise For Duplicate except: # TODO Add Raise For Duplicate
pass pass
@ -158,11 +158,11 @@ def get_asset(asset_id: str, connection):
def get_assets(assets_ids: list, connection) -> list: def get_assets(assets_ids: list, connection) -> list:
_returned_data = [] _returned_data = []
space = connection.space("assets") space = connection.space("assets")
for _id in assets_ids: for _id in list(set(assets_ids)):
asset = space.select(_id, index="assetid_search") asset = space.select(_id, index="assetid_search")
asset = asset.data[0] asset = asset.data[0]
_returned_data.append({"id": asset[0], "data": asset[1]}) _returned_data.append({"id": asset[0], "data": asset[1]})
return _returned_data return sorted(_returned_data, key=lambda k: k["id"], reverse=False)
# @register_query(LocalMongoDBConnection) # @register_query(LocalMongoDBConnection)

View File

@ -66,7 +66,8 @@ def test_write_assets():
def test_get_assets(): def test_get_assets():
from planetmint.backend import connect, query from planetmint.backend import connect
from planetmint.backend.tarantool import query
conn = connect() conn = connect()
assets = [ assets = [