def _populate_database():
get_user_model().objects.create_user(username='******', password='******')
get_user_model().objects.create_superuser(username='******', password='******')
image_name = 'test-image.jpg'
image_bytes = None
with open(CURRENT_DIR + '/' + image_name, 'rb') as image:
image_bytes = image.read()
resource = Resource(
external_id='recpbs29kfas9i',
url='https://test.test/resource.pdf'
)
resource.image.save(image_name, ContentFile(image_bytes), save=True)
resource.save()
for external_id in ('recZxlcM61qaDoOkc', 'recYK5ljTyL3b18J3', 'recvSDrARAcmKogbD'):
practice = Practice(
external_id=external_id,
main_resource=resource,
)
practice.image.save(image_name, ContentFile(image_bytes), save=True)
practice.save()
for category_id in ('rec82929kfas9i', 'rec0098afaooka', 'recppasf09aii'):
category = Category(
external_id=category_id,
practice_external_ids=['recZxlcM61qaDoOkc']
)
category.image.save(image_name, ContentFile(image_bytes), save=True)
category.save()
category.practices.add(Practice.objects.filter(external_id='recZxlcM61qaDoOkc').first())
def _import_categories(self, categories):
"""
Import categories and build mapping of category id (model identifier from export) to (possibly created)
database category object.
"""
categories_to_create = []
for identifier, category in categories.items():
try:
db_category = Category.objects.get(model_identifier=identifier)
except Category.DoesNotExist:
db_category = Category(model_identifier=identifier, name=category["name"],
description=category["description"], color=category["color"])
categories_to_create.append(db_category)
self._mappings.category_mapping[identifier] = db_category
Category.objects.bulk_create(categories_to_create)
self.statistics.categories_created = len(categories_to_create)
def post(self):
parser = reqparse.RequestParser()
parser.add_argument('name', required=True)
parser.add_argument('color', required=False)
try:
data = parser.parse_args()
#maybe add a check to prevent the same exact category from being added twice for one user
new_cat = Category(owner=get_jwt_identity(),
name=data['name'],
color=data['color'])
db.session.add(new_cat)
db.session.commit()
output = {
"id": new_cat.id,
"name": new_cat.name,
"color": new_cat.color
}
return json.dumps(output), 200
except Exception:
raise Exception
ret.append(q.slug)
return ret
# If the workflow does not exist, we inform the user
except ObjectDoesNotExist:
return "Workflow with slug '" + workflowslug + "' does not exist"
# Get or create categories category 1 and category 2
for i in range(1, 3):
print "Creating category " + str(i) + "..."
try:
# Try to find the category
c = Category.objects.get(name="category " + str(i))
except ObjectDoesNotExist:
# If it does not exist, we create it
c = Category(name="category " + str(i))
c.save()
# Get or create workflows with names workflow xy, x in {1,2} and y in {1,2,3} and relate them to category x
for i in range(1, 4):
for j in range(1, 3):
print "Creating workflow " + str(j) + str(i) + "..."
try:
# Try to find the workflow
w = Workflow.objects.get(name="workflow " + str(j) + str(i))
# Relate with category j
w.category.add(Category.objects.get(name="category " + str(j)))
except ObjectDoesNotExist:
# If the workflow does not exist, we create it
w = Workflow(name="workflow " + str(j) + str(i),
json=Command.getJson(populate))
def addCategory(self, baseName):
# create categories
self.createObjects(baseName, 5)
for item in self.data[baseName]:
object = Category(name=item, tooltip="tooltip_" + item)
object.save()
