quicktipp/main.py at master · dedeibel/quicktipp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#!env python3
import argparse

from quicktipp import Quicktipp
from tippgenerator649 import TippGenerator649
from tippgenerator550 import TippGenerator550
from histogram import Histogram, Histogram2d
from journal import write_to_journal, find_in_journal

def test_distribution(quicktipp, n, ignore_blacklist, two_dee):
    quicktipp.set_ignore_blacklist(ignore_blacklist)
    quicktipp.prepare(n)
    if two_dee:
        from histogram import Histogram2d
        histogram = Histogram2d(quicktipp.generator.nums, quicktipp.generator.max_num, quicktipp.generator.print_nums_per_line)
    else:
        from histogram import Histogram
        histogram = Histogram()
    histogram.tipps(quicktipp.get())
    histogram.skipped(quicktipp.get_skipped())
    histogram.show()

def show_distribution(quicktipp, two_dee):
    if two_dee:
        from histogram import Histogram2d
        histogram = Histogram2d(quicktipp.generator.nums, quicktipp.generator.max_num, quicktipp.generator.print_nums_per_line)
    else:
        from histogram import Histogram
        histogram = Histogram()
    histogram.tipps(q.get())
    histogram.skipped(q.get_skipped())
    histogram.show()

if __name__ =='__main__':
    parser = argparse.ArgumentParser(description='Generate good(TM) lotto numbers.')
    parser.add_argument('numbers', metavar='NUMBERS', type=str, nargs=1,
                       help='The amount of numbers you want')
    parser.add_argument('-d', '--dist', dest='show_distribution', action='store_true',
                       help='Additionally shows the number distribution of the generated tips')
    parser.add_argument('-f', '--find', dest='find_numbers', action='store_true',
                       help='Search for the given numbers in the journal four or more matches, comma separated, no spaces as NUMBERS parameter')
    parser.add_argument('-j', '--no-journal', dest='no_journal', action='store_true',
            help='Does not write to journal ~/.quicktipp.log in this run (also does not in -t or -d)')
    parser.add_argument('-p', '--no-prefix', dest='no_prefix', action='store_true',
            help='Prevent showing an index before each tip')
    parser.add_argument('-t', '--test', dest='test_distribution', action='store_true',
                       help='Tests the distribution of n generated and filtered results, use e.g. 10000')
    parser.add_argument('-2', '--two-dee', dest='two_dee', action='store_true', # ;-)
            help='Print distribution histogramm as 2D map of the number coordinates')
    parser.add_argument('-b', '--no-blacklist', dest='no_blacklist', action='store_true',
            help='Ignores the blacklist, kind of useless except for testing')
    parser.add_argument('-e', '--eurojackpot', dest='eurojackpot', action='store_true',
            help='Selects 5 of 50 for eurojackpot')
    parser.add_argument('-v', '--verbose', type=int, default='1',
                       help='Chattyness 0 to 3')

    args = parser.parse_args()
    numbers = args.numbers[0]

    if args.find_numbers:
        find_in_journal(numbers)
        exit(0)

    numbers = int(numbers)

    if args.eurojackpot:
        q = Quicktipp(TippGenerator550())
    else:
        q = Quicktipp(TippGenerator649())

    if args.test_distribution:
        test_distribution(q, numbers, args.no_blacklist, args.two_dee)
        exit(0)

    q.set_ignore_blacklist(args.no_blacklist)
    q.set_verbose(args.verbose)
    q.set_print_index_numbers(not args.no_prefix)
    q.prepare(numbers)
    print(str(q))

    if args.show_distribution:
        show_distribution(q, args.two_dee)
    elif not args.no_journal:
        write_to_journal(q)