Snake case

main
Bob Mottram 2022-01-02 12:32:48 +00:00
parent eb5639ef6e
commit b962b267a0
1 changed files with 44 additions and 44 deletions

View File

@ -38,16 +38,16 @@ from utils import local_actor_url
def e2e_eremove_device(base_dir: str, nickname: str, domain: str, def e2e_eremove_device(base_dir: str, nickname: str, domain: str,
deviceId: str) -> bool: device_id: str) -> bool:
"""Unregisters a device for e2ee """Unregisters a device for e2ee
""" """
personDir = acct_dir(base_dir, nickname, domain) person_dir = acct_dir(base_dir, nickname, domain)
deviceFilename = personDir + '/devices/' + deviceId + '.json' device_filename = person_dir + '/devices/' + device_id + '.json'
if os.path.isfile(deviceFilename): if os.path.isfile(device_filename):
try: try:
os.remove(deviceFilename) os.remove(device_filename)
except OSError: except OSError:
print('EX: e2e_eremove_device unable to delete ' + deviceFilename) print('EX: e2e_eremove_device unable to delete ' + device_filename)
return True return True
return False return False
@ -103,70 +103,70 @@ def e2e_evalid_device(deviceJson: {}) -> bool:
def e2e_eadd_device(base_dir: str, nickname: str, domain: str, def e2e_eadd_device(base_dir: str, nickname: str, domain: str,
deviceId: str, name: str, claimUrl: str, device_id: str, name: str, claim_url: str,
fingerprintPublicKey: str, fingerprint_public_key: str,
identityPublicKey: str, identity_public_key: str,
fingerprintKeyType="Ed25519Key", fingerprint_key_type="Ed25519Key",
identityKeyType="Curve25519Key") -> bool: identity_key_type="Curve25519Key") -> bool:
"""Registers a device for e2ee """Registers a device for e2ee
claimUrl could be something like: claim_url could be something like:
http://localhost:3000/users/admin/claim?id=11119 http://localhost:3000/users/admin/claim?id=11119
""" """
if ' ' in deviceId or '/' in deviceId or \ if ' ' in device_id or '/' in device_id or \
'?' in deviceId or '#' in deviceId or \ '?' in device_id or '#' in device_id or \
'.' in deviceId: '.' in device_id:
return False return False
personDir = acct_dir(base_dir, nickname, domain) person_dir = acct_dir(base_dir, nickname, domain)
if not os.path.isdir(personDir): if not os.path.isdir(person_dir):
return False return False
if not os.path.isdir(personDir + '/devices'): if not os.path.isdir(person_dir + '/devices'):
os.mkdir(personDir + '/devices') os.mkdir(person_dir + '/devices')
deviceDict = { device_dict = {
"deviceId": deviceId, "deviceId": device_id,
"type": "Device", "type": "Device",
"name": name, "name": name,
"claim": claimUrl, "claim": claim_url,
"fingerprintKey": { "fingerprintKey": {
"type": fingerprintKeyType, "type": fingerprint_key_type,
"publicKeyBase64": fingerprintPublicKey "publicKeyBase64": fingerprint_public_key
}, },
"identityKey": { "identityKey": {
"type": identityKeyType, "type": identity_key_type,
"publicKeyBase64": identityPublicKey "publicKeyBase64": identity_public_key
} }
} }
deviceFilename = personDir + '/devices/' + deviceId + '.json' device_filename = person_dir + '/devices/' + device_id + '.json'
return save_json(deviceDict, deviceFilename) return save_json(device_dict, device_filename)
def e2e_edevices_collection(base_dir: str, nickname: str, domain: str, def e2e_edevices_collection(base_dir: str, nickname: str, domain: str,
domain_full: str, http_prefix: str) -> {}: domain_full: str, http_prefix: str) -> {}:
"""Returns a list of registered devices """Returns a list of registered devices
""" """
personDir = acct_dir(base_dir, nickname, domain) person_dir = acct_dir(base_dir, nickname, domain)
if not os.path.isdir(personDir): if not os.path.isdir(person_dir):
return {} return {}
personId = local_actor_url(http_prefix, nickname, domain_full) person_id = local_actor_url(http_prefix, nickname, domain_full)
if not os.path.isdir(personDir + '/devices'): if not os.path.isdir(person_dir + '/devices'):
os.mkdir(personDir + '/devices') os.mkdir(person_dir + '/devices')
deviceList = [] device_list = []
for subdir, dirs, files in os.walk(personDir + '/devices/'): for _, _, files in os.walk(person_dir + '/devices/'):
for dev in files: for dev in files:
if not dev.endswith('.json'): if not dev.endswith('.json'):
continue continue
deviceFilename = os.path.join(personDir + '/devices', dev) device_filename = os.path.join(person_dir + '/devices', dev)
devJson = load_json(deviceFilename) dev_json = load_json(device_filename)
if devJson: if dev_json:
deviceList.append(devJson) device_list.append(dev_json)
break break
devicesDict = { devices_dict = {
'id': personId + '/collections/devices', 'id': person_id + '/collections/devices',
'type': 'Collection', 'type': 'Collection',
'totalItems': len(deviceList), 'totalItems': len(device_list),
'items': deviceList 'items': device_list
} }
return devicesDict return devices_dict
def e2e_edecrypt_message_from_device(message_json: {}) -> str: def e2e_edecrypt_message_from_device(message_json: {}) -> str: