1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# SPDX-License-Identifier: AGPL-3.0-or-later
"""
Manage storage of repository information in KVStore table in database.
"""
import json
from uuid import uuid1
from plinth import kvstore
# kvstore key for repository store
STORAGE_KEY = 'network_storage'
REQUIRED_FIELDS = ['path', 'storage_type', 'added_by_module']
def get_storages(storage_type=None):
"""Get all repositories from store."""
storages = kvstore.get_default(STORAGE_KEY, {})
if storages:
storages = json.loads(storages)
if storage_type:
storages = {
uuid: storage
for uuid, storage in storages.items()
if storage['storage_type'] == storage_type
}
return storages
def get(uuid):
"""Return a repository with given UUID from store."""
storages = get_storages()
return storages[uuid]
def update_or_add(storage):
"""Update an existing or create a new repository in store."""
for field in REQUIRED_FIELDS:
if field not in storage:
raise ValueError('missing storage parameter: %s' % field)
existing_storages = get_storages()
if 'uuid' in storage:
# Replace the existing storage
existing_storages[storage['uuid']] = storage
else:
uuid = str(uuid1())
storage['uuid'] = uuid
existing_storages[uuid] = storage
kvstore.set(STORAGE_KEY, json.dumps(existing_storages))
return storage['uuid']
def delete(uuid):
"""Remove a repository from store."""
storages = get_storages()
del storages[uuid]
kvstore.set(STORAGE_KEY, json.dumps(storages))
|
