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()