plotek.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# pLotek 2.2
# author: Bart Grzybicki <bgrzybicki@gmail.com>
#
# CHANGELOG moved to external file

import random, os, platform, sys, csv, shutil
# importing local modules:
import combinations

try:
    # For Python 3.0 and later
    from urllib.request import urlopen
except ImportError:
    # Fall back to Python 2's urllib2
    from urllib2 import urlopen


def chances(runonce = 0):
    ''' How many combinations are there?
        What is the chance to win any money?
    '''
    win = '? zł'
    if runonce == 0:
        cb = combinations.combinations(49, 6)
        cb = str(cb)
        print('Ilość kombinacji w:\n')
        print('Lotto: ' + combinations.splitthousands(cb, '.'))
        cb = combinations.combinations(42, 5)
        cb = str(cb)
        print('Mini Lotto: ' + combinations.splitthousands(cb, '.'))
        cb = int(combinations.combinations(80, 10) / combinations.combinations(20, 10))
        cb = str(cb)
        print('Multi Multi: ' + combinations.splitthousands(cb, '.'))
        input('Wciśnij ENTER: ')
    running2 = True
    while running2:
        try:
            clearscreen()
            print('Obliczanie prawdopodobieństwa wygranej.')
            print('-------------------------------')
            print('1. Lotto')
            print('2. Mini Lotto')
            print('3. Multi Multi')
            print('q - wyjście do głównego menu')
            game = input('Wybierz grę: ')
            if game == 'q':
                return game
            if game != '' and int(game) in range(1, 4):
                game = int(game)
                if game == 1: # Lotto
                    gamename = 'Lotto'
                    nums = 49
                    start = 3
                    stop = 7
                    numsdrawed = 6
                    running2 = False
                if game == 2: # Mini Lotto
                    gamename = 'Mini Lotto'
                    nums = 42
                    start = 3
                    stop = 6
                    numsdrawed = 5
                    running2 = False
                if game == 3: # Multi Multi
                    gamename = 'Multi Multi'
                    nums = 80
                    running2 = False
                    running3 = True
                    while running3:
                        numsdrawed = input('Podaj ilość typowanych liczb (1 do 10): ')
                        try:
                            if int(numsdrawed) in range(1, 11):
                                numsdrawed = int(numsdrawed)
                                running3 = False
                                if numsdrawed >= 5 and numsdrawed <= 7:
                                    start = 3
                                elif numsdrawed >= 2 and numsdrawed <= 4:
                                    start = 2
                                elif numsdrawed == 1:
                                    start = 1
                                else:
                                    start = 4
                                stop = numsdrawed + 1
                                machinedrawed = 20
                        except:
                            running3 = True
                if game != 3:
                    machinedrawed = numsdrawed
        except:
            print('Wprowadzono błędne dane!')
    clearscreen()
    print('Prawdopodobieństwo trafienia w '+ gamename)
    print('Typowano ' + str(numsdrawed) + ' z ' +str(nums))
    print('-------------------------------')
    running = True
    numsmatched = numsdrawed
    print('Trafiono:')
    while running:
        try:
            if numsdrawed is not None and numsdrawed in range(start, stop):
                if gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 10:
                    win = '250 000 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 9:
                    win = '10 000 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 8:
                    win = '520 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 7:
                    win = '140 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 6:
                    win = '12 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 5:
                    win = '4 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 10 and numsmatched == 4:
                    win = '2 zł'

                if gamename == 'Multi Multi' and numsdrawed == 9 and numsmatched == 9:
                    win = '70 000 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 9 and numsmatched == 8:
                    win = '2 000 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 9 and numsmatched == 7:
                    win = '300 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 9 and numsmatched == 6:
                    win = '42 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 9 and numsmatched == 5:
                    win = '8 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 9 and numsmatched == 4:
                    win = '2 zł'

                if gamename == 'Multi Multi' and numsdrawed == 8 and numsmatched == 8:
                    win = '22 000 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 8 and numsmatched == 7:
                    win = '600 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 8 and numsmatched == 6:
                    win = '60 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 8 and numsmatched == 5:
                    win = '20 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 8 and numsmatched == 4:
                    win = '4 zł'

                if gamename == 'Multi Multi' and numsdrawed == 7 and numsmatched == 7:
                    win = '6 000 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 7 and numsmatched == 6:
                    win = '200 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 7 and numsmatched == 5:
                    win = '20 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 7 and numsmatched == 4:
                    win = '4 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 7 and numsmatched == 3:
                    win = '2 zł'

                if gamename == 'Multi Multi' and numsdrawed == 6 and numsmatched == 6:
                    win = '1 300 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 6 and numsmatched == 5:
                    win = '120 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 6 and numsmatched == 4:
                    win = '8 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 6 and numsmatched == 3:
                    win = '2 zł'

                if gamename == 'Multi Multi' and numsdrawed == 5 and numsmatched == 5:
                    win = '700 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 5 and numsmatched == 4:
                    win = '20 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 5 and numsmatched == 3:
                    win = '4 zł'

                if gamename == 'Multi Multi' and numsdrawed == 4 and numsmatched == 4:
                    win = '84 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 4 and numsmatched == 3:
                    win = '8 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 4 and numsmatched == 2:
                    win = '2 zł'

                if gamename == 'Multi Multi' and numsdrawed == 3 and numsmatched == 3:
                    win = '54 zł'
                elif gamename == 'Multi Multi' and numsdrawed == 3 and numsmatched == 2:
                    win = '2 zł'

                if gamename == 'Multi Multi' and numsdrawed == 2 and numsmatched == 2:
                    win = '16 zł'

                if gamename == 'Multi Multi' and numsdrawed == 1 and numsmatched == 1:
                    win = '4 zł'

                cb = combinations.matched(nums, numsdrawed, numsmatched, machinedrawed)
                cb = str(cb)
                print(str(numsmatched) + ': ' + '1:' + combinations.splitthousands(cb,'.') + ' (' + win + ')')
                numsmatched = numsmatched - 1
                win = '? zł'
                if numsmatched == start - 1:
                    running = False
        except:
            print('Wprowadzono błędne dane!')
    input('Wciśnij ENTER: ')
    clearscreen()
    chances(1)


def comparedbfile(filename):
    ''' Reads the CSV file with space as a delimiter
        and then compares actual draw with historical draws database.
        It also prints matches and counts them.
    '''
    global nums_list, database, matchcount, dbrowcount, dayzero, daylast
    data = []
    database = []
    with open(filename, newline = '\n', encoding = 'utf-8') as csvfile:
        reader = csv.reader(csvfile, delimiter=' ')
        data = [tuple(row) for row in reader]
        index = 0
        for i in data:
            database = data[index][2]
            day = data[index][1]
            dayzero = data[0][1]
            dbrowcount = len(data)
            if index < len(data)-1:
                index += 1
                daylast = data[index][1]
            database = database.split(',')
            list(database)
            database = [int(item) for item in database]
            common_nums = []
            common_nums = frozenset(nums_list).intersection(database) # compare drawed numbers with database
            common_nums = list(common_nums)
            common_nums.sort()
            match = len(common_nums) == len(nums_list)
            if match == True:
                matchcount += 1
    return matchcount, dbrowcount, dayzero, daylast


def geturl(url):
    ''' Gets the URL and saves it to file.

    Full URL must be given as an argument.
    '''
    global workingdir
    full_path = os.path.realpath(__file__)
    workingdir = os.path.dirname(full_path)
    try:
        shutil.copy2(workingdir + '/' + filename, workingdir + '/' + filename + '.orig')
    except Exception:
        pass
    f = open(workingdir + '/' + filename, 'w')
    for line in urlopen(url):
        line = line.decode('utf-8')
        f.write(line)
    f.close()


def dbdownload(gamealias):
    ''' Downloads LOTTO database in text format from mbnet.com.pl

    The argument is gamealias:
    - dl: for Lotto
    - ml: for Mini Lotto
    - mm: for Multi Multi
    '''
    if gamealias == 'dl':
        gamename = 'Lotto'
    elif gamealias == 'ml':
        gamename = 'Mini Lotto'
    else:
        gamename =  'Multi Multi'
    filename = gamealiases[gamealias]
    try:
        print('Pobieranie bazy ' + gamename + '...')
        url = ('http://www.mbnet.com.pl/' + filename)
        geturl(url)
        print('Baza ' + gamename + ' została pobrana :-).')
    except:
        print('Błąd pobierania bazy losowań ' + gamename + '!')
        try:
            shutil.move(workingdir + '/' + filename + '.orig', workingdir + '/' + filename)
        except Exception:
            pass


def alldbdownload():
    ''' Download all LOTTO databases
    '''
    global gamealiases, gamealias, filename
    gamealiases = {'dl': 'dl_razem.txt', 'ml': 'el.txt', 'mm': 'ml.txt'}
    for gamealias, filename in gamealiases.items():
        dbdownload(gamealias)
    input('Wciśnij ENTER: ')
    return


def clearscreen():
    ''' Clears the screen.

    usage: clearscreen()
    '''
    os_platform = platform.system()
    if os_platform == 'Windows':
        os.system('cls')
    else:
        os.system('clear')


def printheader():
    ''' Prints the app header
    '''
    print('**********************************')
    print('* pLotek 2.2          by bartgee *')
    print('**********************************')
    print('u - aktualizacja baz losowań')
    print('p - prawdopodobieństwo wygranej')
    print('1. Lotto')
    print('2. Mini Lotto')
    print('3. Multi Multi')
    print('q - wyjście')
    print('-------------------------------')


def gameselect():
    ''' returns a game name, numbers to draw and number of draws selected by user
    '''
    global gametype, nums_to_draw, number_of_draws, matchcount
    matchcount = 0
    running = True
    while running:
        try:
            gametype = input('Wybierz grę LOTTO (1,2,3 lub ENTER dla LOTTO): ')
            if gametype == 'u':
                running = False
                return gametype
            if gametype == 'p':
                running = False
                return gametype
            if gametype == 'q':
                running = False
                return gametype

            if gametype != '' and int(gametype) in range(1,4):
                gametype = int(gametype)
            if gametype == '':
                gametype = 1
                nums_to_draw = 6
                running = False
            elif gametype == 1 or gametype == 2:
                      nums_to_draw = ''
                      running = False
            elif gametype == 3:
                running = False
                running2 = True
                while running2:
                    try:
                        nums_to_draw = input('Podaj ilość typowanych liczb (1-10 lub ENTER dla 10): ')
                        if nums_to_draw != '' and int(nums_to_draw) in range(1,11):
                            nums_to_draw = int(nums_to_draw)
                        if nums_to_draw == '':
                            nums_to_draw = 10
                            print('Wybrano 10 liczb.')
                            running2 = False
                        elif nums_to_draw in range(1,11):
                            running2 = False
                    except:
                        print('Wprowadzono błędne dane!')
                    running = False
        except:
            print('Wprowadzono błędne dane!')
    running = True
    if gametype == 'u' or gametype == 'p' or gametype == 'q':
        return
    while running:
        try:
            number_of_draws = input('Podaj ilość zakładów (od 1 do 20 lub ENTER dla 1): ')
            if number_of_draws != '' and int(number_of_draws) in range(1,21):
                number_of_draws = int(number_of_draws)
            if number_of_draws == '':
                number_of_draws = 1
                print('Wybrano 1 zakład.')
                running = False
            elif number_of_draws in range(1, 21):
                running = False
        except:
                print('Wprowadzono błędne dane!')
    return gametype, nums_to_draw, number_of_draws


def drawnumbers(gametype, nums_to_draw, number_of_draws):
    ''' Draws the numbers in chosen Lotto - a Polish lottery games. At the moment
    threee of them - Lotto, Mini Lotto and Multi Multi

    The two values must be integers.
    '''
    global game_name, gamefile, nums_all, nums_list
    num_start = 1
    if gametype == 1:
        game_name = 'Lotto'
        gamefile = 'dl_razem.txt'
        num_end = 49
        nums_to_draw = 6
    if gametype == 2:
        game_name = 'Mini Lotto'
        gamefile = 'el.txt'
        num_end = 42
        nums_to_draw = 5
    if gametype == 3:
        game_name = 'Multi Multi'
        gamefile = 'ml.txt'
        num_end = 80
    num_range = range(num_start,num_end)
    nums_all = ''
    draw = number_of_draws
    for draw in range(1, draw+1):
        nums_drawed = random.sample(num_range, nums_to_draw)
        nums_list = []
        for num in nums_drawed:
                nums_list.append(num)

        nums_list.sort()
        nums_all = ''
        for item in nums_list:
                nums_all = nums_all + str(item) + ','

        nums_all = nums_all[:-1]
        print('-------------------------------')
        print('Zakład nr ' + str(draw) + ' dla ' + game_name + ': ' + nums_all)
        full_path = os.path.realpath(__file__)
        workingdir = os.path.dirname(full_path)
        comparedbfile(workingdir + '/' + gamefile)
        printmatches()
    return nums_all, nums_list


def restartgame():
    restart = ''
    restart = input('Nowe losowanie? UWAGA - wyniki ostatniego losowania zostaną utracone! (t/n): ')
    if restart == 't' or restart == '':
        main()
    else:
        sys.exit(0)


def draworupdate():
    ''' Checks whether the user chose to update drawing databases
    '''
    if gametype == 'u':
        clearscreen()
        alldbdownload()
        main()
    elif gametype == 'p':
        clearscreen()
        chances()
        main()
    elif gametype == 'q':
        sys.exit(0)


def printmatches():
    ''' Prints matches if they occured in draws.
    '''
    if matchcount > 1:
        print('Liczby te wylosowano już ' + str(matchcount) + ' razy!')
    elif matchcount == 1:
        print('Liczby te wylosowano już ' + str(matchcount) + ' raz!')
    elif matchcount == 0:
        print('Nigdy nie wylosowano takich liczb!')


def printfooter():
    ''' Prints the footer after drawing
    '''
    print('###############################')
    print('Baza ' + game_name + ' zawiera ' + str(dbrowcount) + ' losowań od dnia ' + str(dayzero) + ' r.')
    print('Ostatnie losowanie w bazie jest z dnia ' + str(daylast) + ' r.')


def main():
    clearscreen()
    printheader()
    gameselect()
    draworupdate()
    drawnumbers(gametype, nums_to_draw, number_of_draws)
    printfooter()
    restartgame()

if __name__ == '__main__':
    main()