diff options
Diffstat (limited to 'bidding_data.py')
| -rw-r--r-- | bidding_data.py | 288 |
1 files changed, 168 insertions, 120 deletions
diff --git a/bidding_data.py b/bidding_data.py index 60381c4..0ee025a 100644 --- a/bidding_data.py +++ b/bidding_data.py @@ -2,126 +2,174 @@ import csv, sys, json, glob, re from os import path from bs4 import BeautifulSoup as bs4 -directions = ['W', 'N', 'E', 'S'] - -def csv_to_list(file_path): - file_data = [] - with file(file_path) as csv_file: - csv_data = csv.reader(csv_file) - for line in csv_data: - file_data.append(line) - return file_data - -def format_bidding(bidding): - html_output = '<table>' - html_output = html_output + '<tr>' - for dir in directions: - html_output = html_output + '<th>' + dir + '</th>' - html_output = html_output + '</tr>' - for bid_round in bidding: +class JFRBidding: + + # alignment of the bidding table + __directions = ['W', 'N', 'E', 'S'] + + # converts csv file to list of row lists + def __csv_to_list(self, file_path): + file_data = [] + with file(file_path) as csv_file: + csv_data = csv.reader(csv_file) + for line in csv_data: + file_data.append(line) + return file_data + + # converts CSV lineup data to {round}.{sector}_{table}.{pair numbers} structure + def __parse_lineup_data(self, sitting_data): + round_lineups = {} + for sitting in sitting_data[1:]: + round_no = int(sitting[2]) + table_no = sitting[0] + '_' + sitting[1] + if not round_lineups.has_key(round_no): + round_lineups[round_no] = {} + round_lineups[round_no][table_no] = sorted([int(sitting[3]), int(sitting[4])]) + return round_lineups + + # converts CSV bidding to {board}.{sector}_{table}.{round}.{bidding}[] structure, + # including erased calls + def __parse_bidding_data(self, bidding_data): + bids = {} + for bid in bidding_data[1:]: + board_no = int(bid[4]) + round_no = int(bid[3]) + table_no = bid[1] + '_' + bid[2] + bid_counter = int(bid[5]) + bid_erased = int(bid[10]) + if not bids.has_key(board_no): + bids[board_no] = {} + if not bids[board_no].has_key(table_no): + bids[board_no][table_no] = {} + if not bids[board_no][table_no].has_key(round_no): + bids[board_no][table_no][round_no] = {} + if bid_erased == 1: + if bids[board_no][table_no][round_no].has_key(bid_counter): + if bids[board_no][table_no][round_no][bid_counter]['direction'] == bid[6]: + bids[board_no][table_no][round_no].pop(bid_counter, None) + else: + bids[board_no][table_no][round_no][bid_counter] = {'direction': bid[6], 'bid': bid[7] } + return bids + + # converts bidding data into HTML table + def __format_bidding(self, bidding): + html_output = '<table>' html_output = html_output + '<tr>' - for bid in bid_round: - bid_match = re.match(r'(\d)([SHDCN])', bid) - if bid_match: - bid = bid_match.group(1) + '<img src="images/' + bid_match.group(2) + '.gif" />' - html_output = html_output + '<td>' + bid + '</td>' + for dir in self.__directions: + html_output = html_output + '<th>' + dir + '</th>' html_output = html_output + '</tr>' - html_output = html_output + '</table>' - return html_output - -def parse_lineup_data(sitting_data): - round_lineups = {} - for sitting in sitting_data[1:]: - round_no = int(sitting[2]) - table_no = sitting[0] + '_' + sitting[1] - if not round_lineups.has_key(round_no): - round_lineups[round_no] = {} - round_lineups[round_no][table_no] = sorted([int(sitting[3]), int(sitting[4])]) - return round_lineups - -def parse_bidding_data(bidding_data): - bids = {} - for bid in bidding_data[1:]: - board_no = int(bid[4]) - round_no = int(bid[3]) - table_no = bid[1] + '_' + bid[2] - bid_counter = int(bid[5]) - bid_erased = int(bid[10]) - if not bids.has_key(board_no): - bids[board_no] = {} - if not bids[board_no].has_key(table_no): - bids[board_no][table_no] = {} - if not bids[board_no][table_no].has_key(round_no): - bids[board_no][table_no][round_no] = {} - if bid_erased == 1: - if bids[board_no][table_no][round_no].has_key(bid_counter): - if bids[board_no][table_no][round_no][bid_counter]['direction'] == bid[6]: - bids[board_no][table_no][round_no].pop(bid_counter, None) + for bid_round in bidding: + html_output = html_output + '<tr>' + for bid in bid_round: + bid_match = re.match(r'(\d)([SHDCN])', bid) + if bid_match: + bid = bid_match.group(1) + '<img src="images/' + bid_match.group(2) + '.gif" />' + html_output = html_output + '<td>' + bid + '</td>' + html_output = html_output + '</tr>' + html_output = html_output + '</table>' + return html_output + + # returns file path for bidding HTML output + def __get_bidding_file_output_path(self, board_no, round_no=None, table_no=None, pair_numbers=None): + return '{0}_bidding_{1:03}_{2}.txt'.format(self.__tournament_prefix, + board_no, + '_'.join(map(str, self.__round_lineups[round_no][table_no] if pair_numbers is None else pair_numbers))) + + def __map_board_numbers(self, custom_mapping=None): + self.__tournament_files = [f for f + in glob.glob(self.__tournament_prefix + '*.html') + if re.search(self.__tournament_files_match, f)] + if custom_mapping is not None: + custom_files = [] + for jfr_number in range(custom_mapping[0], custom_mapping[1]+1): + board_files = [f for f in self.__tournament_files if f.endswith('{0:03}.html'.format(jfr_number))] + if len(board_files): + self.__board_number_mapping[jfr_number - custom_mapping[0] + custom_mapping[2]] = jfr_number + custom_files = custom_files + board_files + self.__tournament_files = custom_files else: - bids[board_no][table_no][round_no][bid_counter] = {'direction': bid[6], 'bid': bid[7] } - return bids - -bidding_data = csv_to_list(sys.argv[1]) -sitting_data = csv_to_list(sys.argv[2]) -round_lineups = parse_lineup_data(sitting_data) -bids = parse_bidding_data(bidding_data) - -tournament_path_prefix = sys.argv[3] + '.html' -output_path = path.dirname(tournament_path_prefix) -tournament_prefix = path.splitext(path.realpath(tournament_path_prefix))[0] - -for board_no, board_data in bids.items(): - for table_no, table_data in board_data.items(): - for round_no, round_data in table_data.items(): - bidding = sorted(round_data) - dealer = round_data[bidding[0]]['direction'] - bidding_table = [[], [], [], []] - for bid_index in bidding: - bid = round_data[bid_index] - bidding_table[directions.index(bid['direction'])].append(bid['bid']) - last_bidder = bid['direction'] - for pos in range(0, directions.index(dealer)): - bidding_table[pos].insert(0, '') - for pos in range(directions.index(last_bidder), len(directions)): - bidding_table[pos].append('') - bidding_table = map(list, zip(*bidding_table)) - # TODO: file_number, a nie deal_number - bidding_file_path = tournament_prefix + '_bidding_' + str(board_no) + '_' + '_'.join(map(str, round_lineups[round_no][table_no])) + '.txt' - with file(bidding_file_path, 'w') as bidding_file: - bidding_file.write(format_bidding(bidding_table)) - -tournament_files_match = re.compile(re.escape(tournament_prefix) + '([0-9]{3})\.html') -tournament_files = [f for f in glob.glob(tournament_prefix + '*.html') if re.search(tournament_files_match, f)] - -deal_numbers = {} - -for tournament_file in tournament_files: - file_number = re.match(tournament_files_match, tournament_file).group(1) - with file(tournament_file, 'r+') as board_html: - board_content = bs4(board_html, from_encoding='utf-8') - header_scripts = board_content.select('head script') - jquery_scripts = [script for script in header_scripts if script['src'] == 'javas/jquery.js'] - if not len(jquery_scripts): - board_content.head.append(bs4('<script src="javas/jquery.js" type="text/javascript"></script>').script) - bidding_scripts = [script for script in header_scripts if script['src'] == 'javas/bidding.js'] - if not len(bidding_scripts): - board_content.head.append(bs4('<script src="javas/bidding.js" type="text/javascript"></script>').script) - board_number = board_content.select('h4')[0].contents[0].strip().replace('ROZDANIE ', '') - deal_numbers[file_number] = board_number - board_html.seek(0) - board_html.write(board_content.prettify('utf-8', formatter='html')) - board_text_path = path.splitext(tournament_file)[0] + '.txt' - with file(board_text_path, 'r+') as board_text: - board_text_content = bs4(board_text, from_encoding='iso-8859-2') - for row in board_text_content.select('tr'): - cells = row.select('td') - if len(cells) == 11: - pair_numbers = sorted([int(cells[1].contents[0]), int(cells[2].contents[0])]) - bidding_link = bs4('<a href="#" class="biddingLink">[lic]</a>') - # TODO: file_number, a nie deal_number - bidding_link.a['data-bidding-link'] = path.basename(tournament_prefix) + '_bidding_' + str(deal_numbers[file_number]) + '_' + '_'.join(map(str, pair_numbers)) + '.txt' - for link in cells[3].select('a.biddingLink'): - link.extract() - cells[3].append(bidding_link) - board_text.seek(0) - board_text.write(board_text_content.table.prettify('iso-8859-2', formatter='html')) + for tournament_file in self.__tournament_files: + file_number = re.match(self.__tournament_files_match, tournament_file).group(1) + with file(tournament_file, 'r+') as board_html: + board_content = bs4(board_html, from_encoding='utf-8') + board_number = board_content.select('h4')[0].contents[0].strip().replace('ROZDANIE ', '') + self.__board_number_mapping[int(board_number)] = int(file_number, 10) + + __round_lineups = {} + __bids = {} + + __tournament_prefix = '' + __tournament_files_match = None + __tournament_files = [] + + # BWS number -> JFR number + __board_number_mapping = {} + + def __init__(self, bidding_file, lineup_file, file_prefix, board_mapping): + self.__round_lineups = self.__parse_lineup_data(self.__csv_to_list(lineup_file)) + self.__bids = self.__parse_bidding_data(self.__csv_to_list(bidding_file)) + self.__tournament_prefix = path.splitext(path.realpath(file_prefix + '.html'))[0] + self.__tournament_files_match = re.compile(re.escape(self.__tournament_prefix) + '([0-9]{3})\.html') + self.__map_board_numbers(board_mapping) + + def write_bidding_tables(self): + for board_no, board_data in self.__bids.items(): + if board_no in self.__board_number_mapping: + for table_no, table_data in board_data.items(): + for round_no, round_data in table_data.items(): + bidding = sorted(round_data) + dealer = round_data[bidding[0]]['direction'] + bidding_table = [[], [], [], []] + for bid_index in bidding: + bid = round_data[bid_index] + bidding_table[self.__directions.index(bid['direction'])].append(bid['bid']) + last_bidder = bid['direction'] + for pos in range(0, self.__directions.index(dealer)): + bidding_table[pos].insert(0, '') + for pos in range(self.__directions.index(last_bidder), len(self.__directions)): + bidding_table[pos].append('') + bidding_table = map(list, zip(*bidding_table)) + bidding_file_path = self.__get_bidding_file_output_path(self.__board_number_mapping[board_no], round_no, table_no) + with file(bidding_file_path, 'w') as bidding_file: + bidding_file.write(self.__format_bidding(bidding_table)) + + def write_bidding_scripts(self): + for tournament_file in self.__tournament_files: + with file(tournament_file, 'r+') as board_html: + board_content = bs4(board_html, from_encoding='utf-8') + header_scripts = board_content.select('head script') + jquery_scripts = [script for script in header_scripts if script['src'] == 'javas/jquery.js'] + if not len(jquery_scripts): + board_content.head.append(bs4('<script src="javas/jquery.js" type="text/javascript"></script>').script) + bidding_scripts = [script for script in header_scripts if script['src'] == 'javas/bidding.js'] + if not len(bidding_scripts): + board_content.head.append(bs4('<script src="javas/bidding.js" type="text/javascript"></script>').script) + board_html.seek(0) + board_html.write(board_content.prettify('utf-8', formatter='html')) + board_html.truncate() + + def write_bidding_links(self): + for tournament_file in self.__tournament_files: + file_number = re.match(self.__tournament_files_match, tournament_file).group(1) + board_text_path = path.splitext(tournament_file)[0] + '.txt' + with file(board_text_path, 'r+') as board_text: + board_text_content = bs4(board_text, from_encoding='iso-8859-2') + for row in board_text_content.select('tr'): + cells = row.select('td') + if len(cells) == 11: + pair_numbers = sorted([int(cells[1].contents[0]), int(cells[2].contents[0])]) + bidding_link = bs4('<a href="#" class="biddingLink">[lic]</a>') + bidding_link.a['data-bidding-link'] = path.basename(self.__get_bidding_file_output_path( + int(file_number, 10), + pair_numbers=pair_numbers + )) + for link in cells[3].select('a.biddingLink'): + link.extract() + cells[3].append(bidding_link) + board_text.seek(0) + board_text.write(board_text_content.table.prettify('iso-8859-2', formatter='html')) + +bidding_parser = JFRBidding(sys.argv[1], sys.argv[2], sys.argv[3], map(int, sys.argv[4:]) if len(sys.argv) > 4 else None) +bidding_parser.write_bidding_tables() +bidding_parser.write_bidding_scripts() +bidding_parser.write_bidding_links() |