import factory

from blog.models import Topic, Blogpost, Tag


class TopicFactory(factory.Factory):
    class Meta:
        model = Topic

    name = factory.Faker("word")
    numbered = False


class PostFactory(factory.Factory):
    class Meta:
        model = Blogpost

    title = factory.Faker("word")
    tags = Tag.objects.all()[0] if len(Tag.objects.all()) != 0 else None
    if len(Topic.objects.all()) == 0:
        TopicFactory.create_batch(10)
    topics = Topic.objects.all()[0]

    mdfile = "/tmp/test.md"