experiments.py

import argparse, sys
from main import Handler
from collections import defaultdict


def count_tags(handler):
    tags = defaultdict(int)

    for photos in handler.photos:
        for tag in photos.tags:
            tags[tag] += 1

    values = sorted([(value, key) for (key, value) in tags.items()], reverse=True)

    return values

def group_quantities(pairs):
    c = defaultdict(int)

    for pair in pairs:
        c[pair[0]] += 1

    return sorted(c.items(), reverse=True)

def not_useful(handler):
    c = 0
    for photo in handler.photos:
        if photo.horizontal and (photo.tags) == 1:
            c += 1

    return c

filenames = ["a_example.txt", "b_lovely_landscapes.txt", "c_memorable_moments.txt", "d_pet_pictures.txt", "e_shiny_selfies.txt"]

if __name__ == "__main__":

    for filename in filenames:
        print("\n" + filename + "\n")
        filename = "data/" + filename

        handler = Handler(filename)

        tag_count = count_tags(handler)
        print(count_tags(handler)[:100])

        groups = group_quantities(tag_count)
        print(groups[:100])

        #not_useful_count = len(list(filter(lambda x: x[0] == 1, groups)))
        print("not useful: %d" % not_useful(handler))
        print("tags count: %d" % len(tag_count))
        print("photos count: %d" % len(handler.photos))
        print("hphotos count: %d" % len(handler.hphotos))
        print("vphotos count: %d" % len(handler.vphotos))