aboutsummaryrefslogtreecommitdiffstats
path: root/pyblackbird_cc/resources/factories.py
diff options
context:
space:
mode:
authorMatthew Lemon <y@yulqen.org>2024-10-15 21:01:31 +0100
committerMatthew Lemon <y@yulqen.org>2024-10-15 21:01:31 +0100
commiteeaddb27560d723ca7d61359744ceb2709fccd2d (patch)
tree04ddbc49ae7b73d5f5a9e1716d7227aecd3b9f85 /pyblackbird_cc/resources/factories.py
parent7a3044c859043837e6c7c95bb4894d04e9b2cbc2 (diff)
Renamed from pyblackbird_cc to alphabetlearning - everywhere
Diffstat (limited to 'pyblackbird_cc/resources/factories.py')
-rw-r--r--pyblackbird_cc/resources/factories.py83
1 files changed, 0 insertions, 83 deletions
diff --git a/pyblackbird_cc/resources/factories.py b/pyblackbird_cc/resources/factories.py
deleted file mode 100644
index fef8cfb..0000000
--- a/pyblackbird_cc/resources/factories.py
+++ /dev/null
@@ -1,83 +0,0 @@
-import itertools
-
-import factory
-
-from .models import PDFPageSnapshot
-from .models import PDFResource
-from .models import Resource
-from .models import ResourceCategory
-from .models import ResourceType
-
-
-class ResourceTypeModelFactory(factory.django.DjangoModelFactory):
- class Meta:
- model = ResourceType
- django_get_or_create = ("name",)
-
- name = factory.Sequence(lambda n: f"Default Resource Type {n}")
-
-
-class ResourceCategoryModelFactory(factory.django.DjangoModelFactory):
- class Meta:
- model = ResourceCategory
- django_get_or_create = ("name",)
-
- name = factory.Sequence(lambda n: f"Default Resource Category {n}")
-
-
-class PDFPageSnapshotModelFactory(factory.django.DjangoModelFactory):
- class Meta:
- model = PDFPageSnapshot
-
- name = factory.Sequence(lambda n: f"pdf_page_snapshot_{n}")
- file_name = factory.Sequence(lambda n: f"pdf_page_snapshot_{n}.jpg")
- pdf_file = factory.SubFactory("pyblackbird_cc.resources.factories.PDFResourceModelFactory")
-
-
-class ResourceModelFactory(factory.django.DjangoModelFactory):
- class Meta:
- model = Resource
-
- name = factory.Sequence(lambda n: f"Default Resource {n}")
- price = factory.Faker("pydecimal", left_digits=4, right_digits=2, positive=True)
- thumbnail_filenames = factory.List(
- [factory.Faker("file_name", extension="jpg") for _ in range(3)]
- )
- resource_type = factory.SubFactory(ResourceTypeModelFactory)
- main_resource_category = factory.SubFactory(ResourceCategoryModelFactory)
- subcategories = factory.RelatedFactoryList(ResourceCategoryModelFactory, size=2)
- description = factory.Faker("paragraph")
- card_description = factory.Faker("text", max_nb_chars=1000)
- age_range = factory.Iterator(["5-7", "7-9", "9-11"])
- curriculum = factory.Iterator(["English", "Scottish", "No curriculum"])
- feature_slot = factory.Iterator(itertools.chain([1, 2, 3], itertools.repeat(None)))
- created_at = factory.Faker("date_time_this_year")
- updated_at = factory.Faker("date_time_this_month")
-
- @factory.post_generation
- def pdfs(self, create, extracted, **kwargs):
- if not create:
- return
-
- if extracted:
- for pdf in extracted:
- self.pdf_resources.add(pdf)
- else:
- PDFResourceModelFactory.create_batch(3, resource=self)
-
-
-class ResourceSubcategoryModelFactory(factory.django.DjangoModelFactory):
- class Meta:
- model = ResourceCategory
- django_get_or_create = ("name",)
-
- name = factory.Sequence(lambda n: f"Default Resource Subcategory {n}")
-
-
-class PDFResourceModelFactory(factory.django.DjangoModelFactory):
- class Meta:
- model = PDFResource
-
- resource = factory.SubFactory(ResourceModelFactory)
- file_name = factory.Sequence(lambda n: f"test_{n}.pdf")
- file_size = factory.Faker("random_int", min=0, max=1000)