from django.db.models import Q


def get_profile(user):
    from euscan_accounts.models import UserProfile
    try:
        return user.get_profile()
    except UserProfile.DoesNotExist:
        UserProfile.objects.create(user=user)
        return user.get_profile()


def get_account_categories(user):
    from djeuscan.models import Package
    # TODO: This is quite ugly
    category_names = [obj.name for obj in get_profile(user).categories.all()]
    return [c for c in Package.objects.categories()
            if c["category"] in category_names]


def get_account_herds(user):
    from djeuscan.models import Package

    ids = [herd.pk for herd in get_profile(user).herds.all()]
    return Package.objects.herds(ids=ids)


def get_account_maintainers(user):
    from djeuscan.models import Package

    ids = [obj.pk for obj in get_profile(user).maintainers.all()]
    return Package.objects.maintainers(ids=ids)


def get_account_versionlogs(profile):
    """
    Returns all watched packages
    """
    from djeuscan.models import Package, VersionLog

    q_categories = Q(category__in=[
        category.name for category in profile.categories.all()])
    q_herds = Q(herds__in=profile.herds.all())
    q_maintainers = Q(maintainers__in=profile.maintainers.all())
    packages = list(profile.packages.all()) + list(Package.objects.filter(
        q_categories | q_herds | q_maintainers))

    overlays = [o.name for o in profile.overlays.all()]

    return VersionLog.objects.filter(
        Q(package__in=packages) | Q(overlay__in=overlays)
    )


def get_user_fav_infos(user):
    upstream_k = lambda c: c["n_versions"] - c["n_packaged"] - c["n_overlay"]

    categories = sorted(get_account_categories(user),
                        key=upstream_k, reverse=True)
    c_upstream = sum([upstream_k(c) for c in categories])
    herds = sorted(get_account_herds(user),
                   key=upstream_k, reverse=True)
    h_upstream = sum([upstream_k(c) for c in herds])
    maintainers = sorted(get_account_maintainers(user),
                         key=upstream_k, reverse=True)
    m_upstream = sum([upstream_k(c) for c in maintainers])
    packages = sorted(
        get_profile(user).packages.all(),
        key=lambda p: p.n_versions - p.n_packaged - p.n_overlay,
        reverse=True
    )
    p_upstream = sum(
        [c.n_versions - c.n_packaged - c.n_overlay for c in packages]
    )

    return {
        "categories": categories, "categories_upstream": c_upstream,
        "herds": herds, "herds_upstream": h_upstream,
        "maintainers": maintainers, "maintainers_upstream": m_upstream,
        "packages": packages, "packages_upstream": p_upstream,
    }