1 files changed, 124 insertions, 0 deletions
diff --git a/core/mongodb/prefabs.py b/core/mongodb/prefabs.py
new file mode 100755
index 00000000..f6f46cbc
--- /dev/null
+++ b/core/mongodb/prefabs.py
@@ -0,0 +1,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})
+
+