def __init__(self, database_url): parsed_url = urlparse(database_url) if 'mongo' in parsed_url.scheme: if 'mongodb' == parsed_url.scheme: client = MongoClient(parsed_url.hostname, parsed_url.port) client.drop_database('cortex_db') elif 'mockmongo' == parsed_url.scheme: client = mongomock.MongoClient(parsed_url.hostname, parsed_url.port, connect=True) enable_gridfs_integration() else: exit(f"Unknown DB {parsed_url.scheme}") self.db = client.cortex_db self.fs = gridfs.GridFS(self.db) self.db.users.create_index([("user_id", ASCENDING)], unique=True) self.db.snapshots.create_index([("user_id", ASCENDING), ("datetime", ASCENDING)]) else: exit(f"Unknown DB {parsed_url.scheme}")
import mongoengine from datetime import datetime from mongomock import gridfs gridfs.enable_gridfs_integration() mongoengine.connect("graphene-mongo-test", host="mongomock://localhost", alias="default") # mongoengine.connect('graphene-mongo-test', host='mongodb://localhost/graphene-mongo-dev') class Publisher(mongoengine.Document): meta = {"collection": "test_publisher"} name = mongoengine.StringField() @property def legal_name(self): return self.name + " Inc." def bad_field(self): return None class Editor(mongoengine.Document): """ An Editor of a publication. """ meta = {"collection": "test_editor"} id = mongoengine.StringField(primary_key=True)
import beer_garden.db.mongo.models import beer_garden.events import beer_garden.requests import beer_garden.router from beer_garden.api.http.authentication import issue_token_pair from beer_garden.db.mongo.models import ( Garden, RawFile, Request, Role, RoleAssignment, System, User, ) enable_gridfs_integration() def format_form_data(metadata: list, data: str, boundary: str): form_data = "\r\n" for item in metadata: form_data += f"{item}\r\n" form_data += f"\r\n{data}" form_data += f"\r\n--{boundary}" return form_data def generate_form_data(system) -> dict: