mirror of https://github.com/D4-project/d4-core
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*-coding:UTF-8 -*
|
||
|
|
|
||
|
|
import os
|
||
|
|
import uuid
|
||
|
|
import redis
|
||
|
|
|
||
|
|
host_redis_metadata = os.getenv('D4_REDIS_METADATA_HOST', "localhost")
|
||
|
|
port_redis_metadata = int(os.getenv('D4_REDIS_METADATA_PORT', 6380))
|
||
|
|
|
||
|
|
r_serv_db = redis.StrictRedis(
|
||
|
|
host=host_redis_metadata,
|
||
|
|
port=port_redis_metadata,
|
||
|
|
db=0)
|
||
|
|
|
||
|
|
def is_valid_uuid_v4(UUID):
|
||
|
|
UUID = UUID.replace('-', '')
|
||
|
|
try:
|
||
|
|
uuid_test = uuid.UUID(hex=UUID, version=4)
|
||
|
|
return uuid_test.hex == UUID
|
||
|
|
except:
|
||
|
|
return False
|
||
|
|
|
||
|
|
## TODO: add user_id + description
|
||
|
|
def register_sensor(req_dict):
|
||
|
|
sensor_uuid = req_dict.get('uuid', None)
|
||
|
|
hmac_key = req_dict.get('hmac_key', None)
|
||
|
|
# verify uuid
|
||
|
|
if not is_valid_uuid_v4(sensor_uuid):
|
||
|
|
return ({"status": "error", "reason": "Invalid uuid"}, 400)
|
||
|
|
# sensor already exist
|
||
|
|
if r_serv_db.exists('metadata_uuid:{}'.format(sensor_uuid)):
|
||
|
|
return ({"status": "error", "reason": "Sensor already registred"}, 409)
|
||
|
|
|
||
|
|
res = _register_sensor(sensor_uuid, hmac_key, user_id=None, description=None)
|
||
|
|
return res
|
||
|
|
|
||
|
|
|
||
|
|
def _register_sensor(sensor_uuid, secret_key, user_id=None, description=None):
|
||
|
|
r_serv_db.hset('metadata_uuid:{}'.format(sensor_uuid), 'hmac_key', secret_key)
|
||
|
|
if user_id:
|
||
|
|
r_serv_db.hset('metadata_uuid:{}'.format(sensor_uuid), 'description', description)
|
||
|
|
if description:
|
||
|
|
r_serv_db.hset('metadata_uuid:{}'.format(sensor_uuid), 'description', description)
|
||
|
|
return ({'uuid': sensor_uuid}, 200)
|