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()