163 lines
5.3 KiB
Python
163 lines
5.3 KiB
Python
"""
|
|
Assets Class object
|
|
"""
|
|
|
|
import logging
|
|
import requests
|
|
import urllib.parse
|
|
import xmltodict
|
|
|
|
|
|
class Assets():
|
|
"""Assets class"""
|
|
|
|
def __init__(self, base_url, jwt):
|
|
"""
|
|
__init__ Init Assets
|
|
|
|
:param base_url: Base URL for SimpleStorageService API
|
|
:param jwt: JWT Authentication for SimpleStorageService API
|
|
"""
|
|
|
|
self.base_url = base_url
|
|
self.jwt = jwt
|
|
|
|
|
|
def get_list(self, business_key, campaign_key):
|
|
"""
|
|
get Get list of assets for the campaign
|
|
|
|
:param business_key: The Business key
|
|
:param campaign_key: The Campaign key
|
|
"""
|
|
|
|
# Set the headers for the request
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Accept': 'application/json',
|
|
'Authorization': f'Bearer {self.jwt}'
|
|
}
|
|
|
|
args = {
|
|
'business_key': business_key,
|
|
'campaign_key': campaign_key
|
|
}
|
|
|
|
# Get the list URL
|
|
response = requests.get(
|
|
f'{self.base_url}/api/v1/files',
|
|
headers=headers,
|
|
params=args,
|
|
timeout=300
|
|
)
|
|
|
|
# Good response. Get the list
|
|
if response.ok:
|
|
assets = {}
|
|
url_json = response.json()
|
|
|
|
if 'list_url' in url_json:
|
|
|
|
headers = {
|
|
'Content-Type': '',
|
|
'Accept': 'application/xml'
|
|
}
|
|
|
|
list_response = requests.get(
|
|
url_json['list_url'],
|
|
headers=headers,
|
|
timeout=300
|
|
)
|
|
|
|
# The json version of the returned list is an invalid structure and doesn't parse correctly.
|
|
# So, we get an XML structure and convert to a properly structured JSON
|
|
if list_response.ok:
|
|
bucket_items_dict = xmltodict.parse(list_response.content)
|
|
|
|
#assets = {}
|
|
#print(f"ListBucketResult:")
|
|
assets['bucket_name'] = bucket_items_dict['ListBucketResult']['Name']
|
|
assets['key_prefix'] = bucket_items_dict['ListBucketResult']['Prefix']
|
|
# for k,v in bucket_items_dict['ListBucketResult'].items():
|
|
# if k == "Contents":
|
|
# print("Contents:")
|
|
# for i,el in enumerate(v):
|
|
# print(f"\tItem {i}:")
|
|
# for k2,v2 in el.items():
|
|
# print(f"\t {k2}: {v2}")
|
|
# else:
|
|
# print(f"{k} = {v}")
|
|
|
|
asset_list = []
|
|
if 'ListBucketResult' in bucket_items_dict:
|
|
if 'Contents' in bucket_items_dict['ListBucketResult']:
|
|
for asset in bucket_items_dict['ListBucketResult']['Contents']:
|
|
name_parts = asset['Key'].rsplit('%2F')
|
|
asset_list.append({'key': asset['Key'], 'filename': urllib.parse.unquote_plus(name_parts[-1])})
|
|
assets['asset_list'] = asset_list
|
|
|
|
# print("\nassets:")
|
|
# print(f"\tbucket_name: {assets['bucket_name']}")
|
|
# print(f"\t key_prefix: {assets['key_prefix']}")
|
|
# if len(assets['asset_list']) > 0:
|
|
# print(f"\t asset_list:")
|
|
# for i,a in enumerate(assets['asset_list']):
|
|
# print(f"\t\t{i:02d}: {a['filename']}")
|
|
# else:
|
|
# print(f"\t asset_list: (no assets being stored)")
|
|
|
|
# print("")
|
|
|
|
return assets
|
|
|
|
# Log error and return null
|
|
logging.error("[AFCASSETS GET LIST] [%s]", response.text)
|
|
logging.error("[AFCASSETS GET LIST] [%s]", response.request.url)
|
|
return None
|
|
|
|
|
|
def file_copyevent(self, business_key, campaign_key, event_key, src_file, dest_file):
|
|
"""
|
|
file_copyevent: Copy a file from Campaign asset library into an event
|
|
|
|
:param business_key: The Business key
|
|
:param campaign_key: The Campaign key
|
|
:param event_key: The Event key
|
|
:param src_file: The Source filename
|
|
:param dst_file: The Destination filename
|
|
"""
|
|
|
|
# Set the headers for the request
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Accept': 'application/json',
|
|
'Authorization': f'Bearer {self.jwt}'
|
|
}
|
|
|
|
args = {
|
|
'business_key': business_key,
|
|
'campaign_key': campaign_key,
|
|
'event_key': event_key,
|
|
'src_file': src_file,
|
|
'dest_file': dest_file
|
|
}
|
|
|
|
print(f"[AFCASSETS]: args = {args}")
|
|
|
|
# Send the copy command
|
|
response = requests.get(
|
|
f'{self.base_url}/api/v1/file/copyevent',
|
|
headers=headers,
|
|
params=args,
|
|
timeout=300
|
|
)
|
|
|
|
# Good response. Get the list
|
|
if response.ok:
|
|
return response.content
|
|
|
|
# Log error and return null
|
|
logging.error("[AFCASSETS FILE_COPYEVENT] [%s]", response.text)
|
|
logging.error("[AFCASSETS FILE_COPYEVENT] [%s]", response.request.url)
|
|
return None
|