diff options
author | 2019-11-11 15:28:57 -0500 | |
---|---|---|
committer | 2019-12-16 11:02:15 -0500 | |
commit | 77900bb3097491cd9fca964c111ea70724e53989 (patch) | |
tree | b23fb51b9e3c465bfcecaede1f0450acfe22e880 /mod/distributorapi/distributor/data_access.py | |
parent | c698e66797bad69b4c77b26b487bf8322989beb0 (diff) |
Add distributor api projectdev-mod
Issue-ID: DCAEGEN2-1860
Signed-off-by: Michael Hwang <mhwang@research.att.com>
Change-Id: I67aa9178b1b1830e330ca1259e8f6b30202945df
Diffstat (limited to 'mod/distributorapi/distributor/data_access.py')
-rw-r--r-- | mod/distributorapi/distributor/data_access.py | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/mod/distributorapi/distributor/data_access.py b/mod/distributorapi/distributor/data_access.py new file mode 100644 index 0000000..e1a45e3 --- /dev/null +++ b/mod/distributorapi/distributor/data_access.py @@ -0,0 +1,89 @@ +# ============LICENSE_START======================================================= +# Copyright (c) 2019 AT&T Intellectual Property. All rights reserved. +# ================================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============LICENSE_END========================================================= +"""Data layer""" + +from datetime import datetime +import uuid + +# TODO: Use real storage +_cache = [] + + +def get_distribution_targets(): + global _cache + return _cache + + +def get_distribution_target(ds_id): + global _cache + result = [ i for i in _cache if i["dt_id"] == ds_id ] + return result[0] if result else {} + +def transform_request(req): + """Transform request to object to store + + NOTE: This method is not safe + """ + ts = datetime.utcnow().isoformat() + req["created"] = ts + req["modified"] = ts + req["dt_id"] = str(uuid.uuid4()) + req["processGroups"] = [] + return req + +def add_distribution_target(dt): + global _cache + _cache.append(dt) + return dt + + +def merge_request(dt, req): + dt["name"] = req["name"] + dt["runtimeApiUrl"] = req["runtimeApiUrl"] + dt["description"] = req.get("description", None) + dt["nextDistributionTargetId"] = req.get("nextDistributionTargetId", None) + dt["modified"] = datetime.utcnow().isoformat() + return dt + +def update_distribution_target(updated_dt): + dt_id = updated_dt["dt_id"] + global _cache + # Did not use list comprehension blah blah because could not do the "return + # True" easily + for i, dt in enumerate(_cache): + if dt["dt_id"] == dt_id: + _cache[i] = updated_dt + return True + return False + + +def delete_distribution_target(dt_id): + global _cache + num_prev = len(_cache) + _cache = list(filter(lambda e: e["dt_id"] != dt_id, _cache)) + return len(_cache) < num_prev + + +def add_process_group(ds_id, process_group): + global _cache + for dt in _cache: + if dt["dt_id"] == ds_id: + process_group["processed"] = datetime.utcnow().isoformat() + dt["processGroups"].append(process_group) + return process_group + return None + + |