metadata_update_test.py

from citadel import Citadel
import json
import arrow
import traceback
import sys
import pdb
import urllib3
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

def eval_res(res, task_name):
    if res:
        print('Success: {0}'.format(task_name))
    else:
        print('Failed: {0}'.format(task_name))
        sys.exit()

with open('config/citadel_config_local.json', 'r') as fp:
    config = json.load(fp)
apikey = config['apikey']
citadel = Citadel(config['hostname'], apikey, False)
point = config['point']

# Query Test
res = citadel.query_points({
    'name': point['name']
})
uuid = res[0]
if uuid:
    print('Success: Point query')
else:
    print('Failed: Point query due to {0}')

# Upsert Query with adding an owner
point['owner'] = 'bk7749@gmail.com'
res = citadel.upsert_metadata(uuid, point)
eval_res(res, 'upsert metadata')




secondary_apikey = config['secondary_apikey']
citadel = Citadel(config['hostname'], secondary_apikey, False)

# Post Data Test
datapoint = config['datapoint']
t = datapoint['timestamp']
datapoint['uuid'] = uuid
data = [datapoint]
res = citadel.post_data(data)
assert res
if res:
    print('Success: post data')
else:
    print('Failed: post data')

# Get Data from a UUID
data = citadel.get_data(uuid, None, None)
if data:
    print("Success: retrieve data only with a UUID")
else:
    print('Failed: empty result')

begin_time = t - 1000000
end_time = t + 1000000
min_lng = -120
min_lat = 30
max_lng = -110
max_lat = 35
uuids = []
res = citadel.query_bbox(begin_time, end_time, min_lng, 
                         min_lat, max_lng, max_lat, uuids)
if res:
    print("Success: retrieve data with BBOX query")
else:
    print('Failed')