import random
from string import letters
from datetime import datetime
from collections import defaultdict

import factory

from djeuscan.models import Herd, Maintainer, Package, Version, EuscanResult


def random_string(length=10):
    return "".join([random.choice(letters) for _ in range(length)])


class HerdFactory(factory.Factory):
    FACTORY_FOR = Herd

    herd = factory.LazyAttribute(lambda a: random_string())
    email = factory.LazyAttribute(lambda a: "%s@example.com" % a.herd)


class MaintainerFactory(factory.Factory):
    FACTORY_FOR = Maintainer

    name = factory.LazyAttribute(lambda a: random_string())
    email = factory.LazyAttribute(lambda a: "%s@example.com" % a.name)


class PackageFactory(factory.Factory):
    FACTORY_FOR = Package

    category = factory.LazyAttribute(lambda a: random_string())
    name = factory.LazyAttribute(lambda a: random_string())
    description = "This is a test package"
    homepage = "http://testpackage.com"


class VersionFactory(factory.Factory):
    FACTORY_FOR = Version

    package = factory.LazyAttribute(lambda a: PackageFactory())
    slot = "1"
    revision = "1"
    version = "0.1"
    packaged = True
    overlay = "gentoo"
    urls = "http://packageurl.com"
    alive = True


class EuscanResultFactory(factory.Factory):
    FACTORY_FOR = EuscanResult

    package = factory.LazyAttribute(lambda a: PackageFactory())
    datetime = datetime.now()
    result = "this is the result"


def setup_maintainers():
    maintainers = [MaintainerFactory.create() for _ in range(10)]
    packages = []
    for i in range(0, 10, 2):
        p = PackageFactory.create()
        p.maintainers.add(maintainers[i])
        p.maintainers.add(maintainers[i + 1])
        packages.append(p)
    return maintainers, packages


def setup_categories():
    packages = [PackageFactory.create() for _ in range(10)]
    categories = [p.category for p in packages]
    return categories, packages


def setup_herds():
    herds = [HerdFactory.create() for _ in range(10)]
    packages = []

    for i in range(0, 10, 2):
        p = PackageFactory.create()
        p.herds.add(herds[i])
        p.herds.add(herds[i + 1])
        packages.append(p)
    return herds, packages


def setup_overlays():
    overlays = [random_string() for _ in range(3)]
    packages = defaultdict(list)

    for _ in range(3):
        package = PackageFactory.create()
        for overlay in overlays:
            VersionFactory.create(package=package,
                                  overlay=overlay)
            packages[overlay].append(package)

    return overlays, packages