"""Module saves metadata about uploaded asset.""" import os from oto import response from assets.constants import error from assets.models import asset_upload as asset_upload_model def update_asset_upload( asset_type, filename, upc, product_id, track_unique_id): """Save metadata about uploaded asset into asset_upload table. Args: asset_type (str): Asset type. filename (str): Filename of uploaded asset. upc (str): UPC. product_id (int): Product id. track_unique_id (int): Track unique id. Returns: response.Response: Success or error message. """ clean_filename, ext = os.path.splitext(filename) asset_upload_result = asset_upload_model.get_asset_upload( clean_filename) if not asset_upload_result: return asset_upload_result asset_upload_update_result = asset_upload_model.update_asset_upload( clean_filename, asset_type, upc, track_unique_id, product_id) if not asset_upload_update_result: return asset_upload_update_result return response.Response({'status': error.SUCCESS_CODE})