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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#!/Users/jacobburley/thesis-src/opendc/mongodb/opendc_testing/bin/python3
#Change shebang to /usr/bin/python3 before using with docker
# encoding: utf-8
"""
prefabs
Python Library for interacting with mongoDB prefabs collection.
"""
import urllib.parse
import pprint
import sys
import os
import json
import re
import ujson
#import pyyaml
from pymongo import MongoClient
from bson.json_util import loads, dumps, RELAXED_JSON_OPTIONS, CANONICAL_JSON_OPTIONS
#mongodb_opendc_db = os.environ['OPENDC_DB']
#mongodb_opendc_user = os.environ['OPENDC_DB_USERNAME']
#mongodb_opendc_password = os.environ['OPENDC_DB_PASSWORD']
#if mongodb_opendc_db == None or mongodb_opendc_user == None or mongodb_opendc_password == None:
# print("One or more environment variables are not set correctly. \nYou may experience issues connecting to the mongodb database.")
user = urllib.parse.quote_plus('opendc') #TODO: replace this with environment variable
password = urllib.parse.quote_plus('opendcpassword') #TODO: same as above
database = urllib.parse.quote_plus('opendc')
client = MongoClient('mongodb://%s:%s@localhost/default_db?authSource=%s' % (user, password, database))
opendcdb = client.opendc
prefabs_collection = opendcdb.prefabs
def add(prefab_file, name):
if(re.match(r"\w+(\\\ \w*)*\.json", prefab_file)):
try:
with open(prefab_file, "r") as json_file:
json_prefab = json.load(json_file)
#print(json_prefab)
if name != None:
json_prefab["name"] = name
try:
prefab_id = prefabs_collection.insert(json_prefab)
except ConnectionFailure:
print("ERROR: Could not connect to the mongoDB database.")
except DuplicateKeyError:
print("ERROR: A prefab with the same unique ID already exists in the database. \nPlease remove the '_id' before trying again.\nYour prefab has not been imported.")
except:
print("ERROR: A general error has occurred. Your prefab has not been imported.")
if prefab_id != None:
if name != None:
print(f'Prefab "{name}" has been imported successfully.')
else:
print(f'Prefab "{prefab_file}" has been imported successfully.')
except FileNotFoundError:
print(f"ERROR: {prefab_file} could not be found in the specified path. No prefabs have been imported.")
elif(re.match(r"\w+(\\\ \w*)*\.yml", prefab_file)):
print("expecting a yaml file here")
#yaml
else:
print("The filetype provided is an unsupported filetype.")
#unsupported filetype
def clone(prefab_name, new_name):
bson = prefabs_collection.find_one({'name': prefab_name})
json_string = dumps(bson) #convert BSON representation to JSON
chosen_prefab = json.loads(json_string) #load as a JSON object
chosen_prefab.pop("_id") # clean out our _id field from the export: mongo will generate a new one if this is imported back in
if new_name != None:
chosen_prefab["name"] = new_name
try:
prefab_id = prefabs_collection.insert_one(chosen_prefab)
except ConnectionFailure:
print("ERROR: Could not connect to the mongoDB database.")
except:
print("ERROR: A general error has occurred. Your selected prefab has not been cloned.")
if prefab_id != None:
if new_name != None:
print(f'Prefab "{prefab_name}" has been cloned successfully as {new_name}.')
else:
print(f'Prefab "{prefab_name}" has been cloned successfully.')
def export(prefab_name, type):
bson = prefabs_collection.find_one({'name': prefab_name})
json_string = dumps(bson) #convert BSON representation to JSON
chosen_prefab = json.loads(json_string) #load as a JSON object
chosen_prefab.pop("_id") # clean out our _id field from the export: mongo will generate a new one if this is imported back in
with open(f'{prefab_name}.json', 'w', encoding='utf8') as f:
json.dump(chosen_prefab, f, ensure_ascii=False, indent=4)
print(f'Prefab {prefab_name} written to {os.getcwd()}/{prefab_name}.json.')
#pprint.pprint(json_string)
#pprint.pprint(json.loads(str(json_string)))
def list():
#TODO: why does it output in single quotations?
cursor = prefabs_collection.find()
prefabs = []
for record in cursor:
#pprint.pprint(record)
#print(record)
json_string = dumps(record, json_options=RELAXED_JSON_OPTIONS) ##pymongo retrieves BSON objects, which need to be converted to json for pythons json module
prefabs.append(json.loads(json_string))
#print(f'There are {str(len(prefabs))} prefabs in the database. They are:')
print("Name Author")
for prefab in prefabs:
if(prefab['visibility'] == "private"):
continue
print(f"{prefab['name']} {prefab['author']}")
#pprint.pprint(prefab)
def remove(prefab_name):
prefabs_collection.delete_one({'name': prefab_name})
|
