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:
