diff options
| author | emkael <emkael@tlen.pl> | 2015-10-09 19:55:31 +0200 |
|---|---|---|
| committer | emkael <emkael@tlen.pl> | 2015-10-09 19:55:31 +0200 |
| commit | 388c4a25510d03d006b1caa3e1fd0a4928794662 (patch) | |
| tree | 3cd3fb539693dce191e339b3c94cf373a41dc476 | |
| parent | 1367feca21f425a0ee5a07173e8a1897ec427e20 (diff) | |
* PDF-fix refactor
| -rw-r--r-- | rrb2txt.py | 775 |
1 files changed, 407 insertions, 368 deletions
@@ -9,377 +9,416 @@ from bs4 import BeautifulSoup, Comment from glob import glob
from itertools import chain, cycle
-directory = sys.argv[1] if len(sys.argv) > 1 else '.'
-filepath = os.path.join(directory, 'pary.txt')
-
-content = None
-with open(filepath, 'r') as file_content:
- content = BeautifulSoup(file_content.read())
-
-pdf_fixed = False
-comments = content.findAll(text=lambda t: isinstance(t, Comment))
-for comment in comments:
- if comment == 'fixpdf.py':
- pdf_fixed = True
- break
-
-if not pdf_fixed:
- header = content.select('thead tr')[0]
- body = content.select('tbody tr')
-
- if not header.find_all(text='+/-'):
- tag = content.new_tag('td', style='display:none')
- tag.string = '+/-'
- header.find(lambda n: n.text[0:5] == 'wynik').insert_before(tag)
- for row in body:
- tag = content.new_tag('td', style='display:none', rowspan=2)
- score_cell = row.select('td.right')
- if score_cell:
- score_cell[0].insert_before(tag)
-
- extra_headers = ['PKL', 'PDF', 'nagroda']
- extra_headers_present = [bool(header.find_all(text=h))
- for h in extra_headers]
-
- extra_headers_offset = 8
-
- for i in range(0, len(extra_headers)):
- if not extra_headers_present[i]:
- tag = content.new_tag('td', style='display:none')
- tag.string = extra_headers[i]
- header.select('td')[extra_headers_offset].insert_after(tag)
+
+class RRBPdfFix(object):
+
+ __filepath = ''
+ __content = None
+
+ def __init__(self, filepath):
+ self.__filepath = filepath
+ with open(self.__filepath, 'r') as file_content:
+ content = file_content.read()
+ self.__content = BeautifulSoup(content, 'lxml')
+
+ def check_fixpdf(self):
+ comments = self.__content.findAll(
+ text=lambda t: isinstance(t, Comment))
+ for comment in comments:
+ if comment == 'fixpdf.py':
+ return True
+ return False
+
+ def __fix_adjustements_column(self, header, body):
+ if not header.find_all(text='+/-'):
+ tag = self.__content.new_tag('td', style='display:none')
+ tag.string = '+/-'
+ header.find(lambda n: n.text[0:5] == 'wynik').insert_before(tag)
for row in body:
- cells = row.find_all('td')
- if len(cells) >= extra_headers_offset:
- tag = content.new_tag(
- 'td', style='display:none', rowspan=2)
- cells[extra_headers_offset].insert_after(tag)
- extra_headers_offset += 1
-
- def get_points_count(row):
- try:
- return row.find_all('td')[10].text.count('|')
- except IndexError:
- return 0
-
- max_points_count = max([get_points_count(row) for row in body]) + 1
-
- header.find_all('td')[10]['colspan'] = max_points_count
-
- for row in body:
- cells = row.find_all('td')
- if len(cells) >= extra_headers_offset:
- span = max_points_count
- points = cells[10].text.split('|')
- new_cells = []
- for point in points:
- tag = content.new_tag('td', rowspan=2)
- tag.string = point
- new_cells.append(tag)
- span -= 1
- if span > 0:
- new_cells[-1]['colspan'] = span + 1
- for new_cell in new_cells:
- cells[11].insert_before(new_cell)
- cells[10].extract()
-
- content.body.append(content.new_string('fixpdf.py', Comment))
-
- content.body.p.extract()
- new_content = content.find('body').decode_contents()
- new_length = len(new_content) + 1
-
- output = open(filepath.name, 'wb')
- output.write('%012d' % new_length)
- output.write('\n')
- output.write(new_content.encode('utf-8'))
- output.write('\n')
-
-
-def format_boards(rows):
- rows = rows[1:4]
- header = rows[0][0].split(os.linesep)
- rows[0][0] = '/'.join(reversed(header[1]
- .replace('obie przed', 'NIKT')
- .replace('obie po', 'OBIE')
- .split(' / ')))
- rows[1][1] = ''
-
- def split_hand(hand):
- return hand.split(os.linesep)
-
- rows[0][1] = split_hand(rows[0][1])
- rows[1][0] = split_hand(rows[1][0])
- rows[1][2] = split_hand(rows[1][2])
- rows[2][1] = split_hand(rows[2][1])
-
- def side_rows(row):
- ret = [
- [row[0],
- row[1][0][2:],
- row[2]]
- ]
- for i in range(1, 4):
- ret.append(['',
- row[1][i][2:] or '===',
- ''])
- return ret
-
- def middle_rows(row):
- ret = []
- for i in range(0, 4):
- ret.append([row[0][i][2:] or '===',
- row[1],
- row[2][i][2:] or '==='])
- return ret
-
- rows = side_rows(rows[0]) + middle_rows(rows[1]) + side_rows(rows[2])
- header = 'ROZDANIE NR ' + header[0]
- output = [header, '']
- output.append('{:10s}{:6s}{:10s}'.format(*rows.pop(0)))
- for row in rows:
- output.append(' {:8s}{:6s}{:10s}'.format(*row))
- output.append('')
- return output
-
-
-def format_protocols(rows):
- output = [' ZAPIS WYNIK',
- ' NS EW KONTRAKT WIST NS EW NS EW']
- for row in rows:
- content = []
- if len(row) == 10:
- content = [
- row[0],
- row[1],
- ' ' + row[2] + ' ' + row[3] + ' ' + row[5],
- row[4] or '',
- row[6] or '',
- '-' + row[7] if row[7] else '',
- '{:.1f}'.format(float(row[8])),
- '{:.1f}'.format(float(row[9]))
- ]
- elif len(row) == 9:
- content = [
- row[0],
- row[1],
- ' ' + row[2] + ' ' + row[3] + ' ' + row[5],
- row[4],
- '0',
- '',
- '{:.1f}'.format(float(row[7])),
- '{:.1f}'.format(float(row[8]))
+ tag = self.__content.new_tag(
+ 'td', style='display:none', rowspan=2)
+ score_cell = row.select('td.right')
+ if score_cell:
+ score_cell[0].insert_before(tag)
+
+ def __fix_extra_columns(self, header, body, extra_headers):
+ extra_headers_present = [bool(header.find_all(text=h))
+ for h in extra_headers]
+
+ extra_headers_offset = 8
+
+ for i in range(0, len(extra_headers)):
+ if not extra_headers_present[i]:
+ tag = self.__content.new_tag('td', style='display:none')
+ tag.string = extra_headers[i]
+ header.select('td')[extra_headers_offset].insert_after(tag)
+ for row in body:
+ cells = row.find_all('td')
+ if len(cells) >= extra_headers_offset:
+ tag = self.__content.new_tag(
+ 'td', style='display:none', rowspan=2)
+ cells[extra_headers_offset].insert_after(tag)
+ extra_headers_offset += 1
+ return extra_headers_offset
+
+ def __fix_pdf_columns(self, header, body, extra_headers_offset):
+
+ def get_pdf_count(row):
+ try:
+ return row.find_all('td')[10].text.count('|')
+ except IndexError:
+ return 0
+
+ max_points_count = max([get_pdf_count(row) for row in body]) + 1
+
+ header.find_all('td')[10]['colspan'] = max_points_count
+
+ for row in body:
+ cells = row.find_all('td')
+ if len(cells) >= extra_headers_offset:
+ span = max_points_count
+ points = cells[10].text.split('|')
+ new_cells = []
+ for point in points:
+ tag = self.__content.new_tag('td', rowspan=2)
+ tag.string = point
+ new_cells.append(tag)
+ span -= 1
+ if span > 0:
+ new_cells[-1]['colspan'] = span + 1
+ for new_cell in new_cells:
+ cells[11].insert_before(new_cell)
+ cells[10].extract()
+
+ def __fix_table(self):
+ content = self.__content
+
+ header = content.select('thead tr')[0]
+ body = content.select('tbody tr')
+
+ self.__fix_adjustements_column(header, body)
+
+ extra_headers = ['PKL', 'PDF', 'nagroda']
+ extra_headers_offset = self.__fix_extra_columns(
+ header, body, extra_headers)
+
+ self.__fix_pdf_columns(header, body, extra_headers_offset)
+
+ return content
+
+ def __write_content(self, content):
+ content.body.append(content.new_string('fixpdf.py', Comment))
+ content.body.p.extract()
+ new_content = content.find('body').decode_contents()
+ new_length = len(new_content) + 1
+
+ with open(self.__filepath, 'wb') as output:
+ output.write('%012d' % new_length)
+ output.write('\n')
+ output.write(new_content.encode('utf-8'))
+ output.write('\n')
+
+ def fixpdf(self):
+ if not self.check_fixpdf():
+ self.__write_content(self.__fix_table())
+
+
+class RRBTxtGen(object):
+
+ def format_boards(self, rows):
+ rows = rows[1:4]
+ header = rows[0][0].split(os.linesep)
+ rows[0][0] = '/'.join(reversed(header[1]
+ .replace('obie przed', 'NIKT')
+ .replace('obie po', 'OBIE')
+ .split(' / ')))
+ rows[1][1] = ''
+
+ def split_hand(hand):
+ return hand.split(os.linesep)
+
+ rows[0][1] = split_hand(rows[0][1])
+ rows[1][0] = split_hand(rows[1][0])
+ rows[1][2] = split_hand(rows[1][2])
+ rows[2][1] = split_hand(rows[2][1])
+
+ def side_rows(row):
+ ret = [
+ [row[0],
+ row[1][0][2:],
+ row[2]]
]
- if content:
- output.append(
- u'{:>3s} {:>3s} {:11s}{:^4s}{:>4s}{:>5s} {:>5s} {:>5s}'.format(
- *content))
- elif len(row) != 4 and len(row) != 8:
- print 'protocols: row of unexpected length'
- print row
- output.append('')
- return output
-
-
-def format_results(rows):
- rows.pop(0)
- content = []
- link_regex = re.compile(r'^http://www\.msc\.com\.pl')
- cezar_ids = [
- '{:05d}'.format(int(
- dict(urlparse.parse_qsl(urlparse.urlparse(row.pop()).query))['r']))
- if re.match(link_regex, row[-1])
- else ''
- for row in rows]
- pdf_columns = max([len(row) for row in rows]) - 11
- for row in rows:
- length = len(row)
- if length > 5:
- content.append(row[0:3] + [cezar_ids.pop(0)] + row[3:])
- elif length == 5:
- content.append([''] * 2 + row[0:1] + [
- cezar_ids.pop(0)] + row[1:] + [''] * (3 + pdf_columns))
- elif length == 4:
- if len(row[3]) != 2:
- content.append([''] * 2 + row[0:1] + [cezar_ids.pop(0)] +
- row[1:3] + content[-1][6:7] +
- row[3:4] + [''] * (3 + pdf_columns))
- else:
- content.append([''] * 2 + row[0:1] + [
- cezar_ids.pop(0)] + row[1:4] + [''] * (4 + pdf_columns))
- elif length == 3:
- content.append([''] * 2 + row[0:1] +
- [cezar_ids.pop(0)] + row[1:3] + content[-1][6:8] +
- [''] * (3 + pdf_columns))
- wk_sum = sum([float(c[5]) if len(c[5]) else 0.0 for c in content])
- output = []
- name_column = max([len(r[2]) for r in content])
- output.append('%s %s %s %s %s' % (
- 'M-CE NR',
- ' ' * name_column,
- 'WK CEZAR +/- WYNIK PKL',
- ('{:^' + str(3 * pdf_columns) + 's}').format('PDF'),
- 'NAGRODA'
- ))
- output.append('-' * len(output[-1]))
- for c in content:
- line = (
- u'{:>3s} {:>3s} {:' + unicode(name_column) +
- u's} {:>4s} {:2s} {:5s} {:2s} {:>5s} {:>6s} {:>3s}').format(
- *(c[0:3] + c[5:7] + c[3:5] + c[8:11]))
- pdf = (
- u' {:' + unicode(3 * pdf_columns) + u's}').format(
- ''.join([u'{:>3s}'.format(cc) for cc in c[11:-1]]))
- line += pdf
- line += u' {:>6s}'.format(c[-1])
- output.append(line)
- output.append(' ' * (8 + name_column) + '-----')
- output.append(
- ('{:>' + str(13 + name_column) + 's}').format(
- 'Suma WK = {:.1f}'.format(wk_sum)))
- return output
-
-
-def format_histories(rows):
- header = rows.pop(0)[0]
- rows.pop(0)
- if ' pauza ' in header:
- return []
- output = ['WYNIKI PARY NR ' + header,
- '']
- content_rows = []
- add_separator = False
- for row in rows:
+ for i in range(1, 4):
+ ret.append(['',
+ row[1][i][2:] or '===',
+ ''])
+ return ret
+
+ def middle_rows(row):
+ ret = []
+ for i in range(0, 4):
+ ret.append([row[0][i][2:] or '===',
+ row[1],
+ row[2][i][2:] or '==='])
+ return ret
+
+ rows = side_rows(rows[0]) + middle_rows(rows[1]) + side_rows(rows[2])
+ header = 'ROZDANIE NR ' + header[0]
+ output = [header, '']
+ output.append('{:10s}{:6s}{:10s}'.format(*rows.pop(0)))
+ for row in rows:
+ output.append(' {:8s}{:6s}{:10s}'.format(*row))
+ output.append('')
+ return output
+
+
+ def format_protocols(self, rows):
+ output = [' ZAPIS WYNIK',
+ ' NS EW KONTRAKT WIST NS EW NS EW']
+ for row in rows:
+ content = []
+ if len(row) == 10:
+ content = [
+ row[0],
+ row[1],
+ ' ' + row[2] + ' ' + row[3] + ' ' + row[5],
+ row[4] or '',
+ row[6] or '',
+ '-' + row[7] if row[7] else '',
+ '{:.1f}'.format(float(row[8])),
+ '{:.1f}'.format(float(row[9]))
+ ]
+ elif len(row) == 9:
+ content = [
+ row[0],
+ row[1],
+ ' ' + row[2] + ' ' + row[3] + ' ' + row[5],
+ row[4],
+ '0',
+ '',
+ '{:.1f}'.format(float(row[7])),
+ '{:.1f}'.format(float(row[8]))
+ ]
+ if content:
+ output.append(
+ u'{:>3s} {:>3s} {:11s}{:^4s}{:>4s}{:>5s} {:>5s} {:>5s}'.format(
+ *content))
+ elif len(row) != 4 and len(row) != 8:
+ print 'protocols: row of unexpected length'
+ print row
+ output.append('')
+ return output
+
+
+ def format_results(self, rows):
+ rows.pop(0)
content = []
- if len(row) == 11:
- add_separator = (
- len(''.join(row[0:9])) == 0) and (
- (add_separator is False) or (row[-2] == 'miejsce'))
- content = row[0:4] + [
- row[4] + ' ' + row[5] + ' ' + row[7]
- ] + [row[6]] + row[8:11]
- elif len(row) == 10:
- content = [''] + row[0:3] + [
- row[3] + ' ' + row[4] + ' ' + row[6]
- ] + [row[5]] + row[7:10]
- elif len(row) == 9:
- content = ['', ''] + row[0:2] + [
- row[2] + ' ' + row[3] + ' ' + row[5]
- ] + [row[4]] + row[6:9]
- if content:
- if add_separator:
- content_rows.append(
- ['', '', '', '', '', '', '', '-------', '--------'])
- content_rows.append(content)
- else:
- print 'histories: unexpected row length'
- print row
- column_width = max([len(r[1]) for r in content_rows])
- content_rows = [[
- 'RND', 'PRZECIWNIK', 'RZD', ' ', 'KONTRAKT', 'WIST',
- 'ZAPIS', 'WYNIK ', u'/ BIEŻĄCY'
- ]] + content_rows
- for content in content_rows:
- if content[6]:
- score_align = u'>' if content[6][0] == u'-' else (
- u'' if content[6][0] == u'+' else u'^')
- else:
- score_align = u''
+ link_regex = re.compile(r'^http://www\.msc\.com\.pl')
+ cezar_ids = [
+ '{:05d}'.format(int(
+ dict(urlparse.parse_qsl(urlparse.urlparse(row.pop()).query))['r']))
+ if re.match(link_regex, row[-1])
+ else ''
+ for row in rows]
+ pdf_columns = max([len(row) for row in rows]) - 11
+ for row in rows:
+ length = len(row)
+ if length > 5:
+ content.append(row[0:3] + [cezar_ids.pop(0)] + row[3:])
+ elif length == 5:
+ content.append([''] * 2 + row[0:1] + [
+ cezar_ids.pop(0)] + row[1:] + [''] * (3 + pdf_columns))
+ elif length == 4:
+ if len(row[3]) != 2:
+ content.append([''] * 2 + row[0:1] + [cezar_ids.pop(0)] +
+ row[1:3] + content[-1][6:7] +
+ row[3:4] + [''] * (3 + pdf_columns))
+ else:
+ content.append([''] * 2 + row[0:1] + [
+ cezar_ids.pop(0)] + row[1:4] + [''] * (4 + pdf_columns))
+ elif length == 3:
+ content.append([''] * 2 + row[0:1] +
+ [cezar_ids.pop(0)] + row[1:3] + content[-1][6:8] +
+ [''] * (3 + pdf_columns))
+ wk_sum = sum([float(c[5]) if len(c[5]) else 0.0 for c in content])
+ output = []
+ name_column = max([len(r[2]) for r in content])
+ output.append('%s %s %s %s %s' % (
+ 'M-CE NR',
+ ' ' * name_column,
+ 'WK CEZAR +/- WYNIK PKL',
+ ('{:^' + str(3 * pdf_columns) + 's}').format('PDF'),
+ 'NAGRODA'
+ ))
+ output.append('-' * len(output[-1]))
+ for c in content:
+ line = (
+ u'{:>3s} {:>3s} {:' + unicode(name_column) +
+ u's} {:>4s} {:2s} {:5s} {:2s} {:>5s} {:>6s} {:>3s}').format(
+ *(c[0:3] + c[5:7] + c[3:5] + c[8:11]))
+ pdf = (
+ u' {:' + unicode(3 * pdf_columns) + u's}').format(
+ ''.join([u'{:>3s}'.format(cc) for cc in c[11:-1]]))
+ line += pdf
+ line += u' {:>6s}'.format(c[-1])
+ output.append(line)
+ output.append(' ' * (8 + name_column) + '-----')
output.append(
- (u'{:>3s} {:' + unicode(column_width) +
- u's} {:>3s} {:2s} {:9s}{:^4s} {:' +
- score_align + u'7s} {:>7s}{:>8s}').format(
- *[c or ' ' for c in content]))
- output.insert(3, '-' * len(output[2]))
- output.append('')
- return output
-
-
-def format_rows(rows, rowtype):
- return globals()['format_' + rowtype](rows)
-
-
-def get_rows(content):
- soup = BeautifulSoup(content)
- output = []
- link_regex = re.compile(r'^http://www\.msc\.com\.pl')
- header = soup.find('h2')
- if header:
- output.append([header.text])
- for table_row in soup.find_all('tr'):
- row = map(lambda t:
- os.linesep.join(t.stripped_strings),
- table_row.find_all('td'))
- row = row + map(lambda l:
- l['href'],
- table_row.find_all('a', {'href': link_regex}))
- output.append(row)
- return output
-
-
-def get_content(filepath):
- return re.sub('<img src=".*/(.*).gif" ?/>',
- lambda img: img.group(1)[0].capitalize(),
- open(filepath, 'r').read())
-
-
-def get_header(directory):
- soup = BeautifulSoup(
- open(os.path.join(directory, 'index.html'), 'r').read())
- return [node.text for node in soup.select('#header *')]
-
-
-def get_files(directory):
- return dict(map(lambda (key, val): (
- key,
- reduce(list.__add__, map(
- lambda v: sorted(glob(os.path.join(directory, v))), val), [])),
- {
- 'boards': ['d?.txt', 'd??.txt'],
- 'protocols': ['p?.txt', 'p??.txt'],
- 'histories': ['h?.txt', 'h??.txt'],
- 'results': ['pary.txt'],
- }.items()))
-
-
-def compile_dir(directory):
- files = get_files(directory)
- return dict(
- map(lambda (key, val):
- (
- key,
- list(
- chain(
- *list(
- i.next() for i in cycle(
- map(lambda v:
- iter(
- map(lambda file:
- format_rows(
- get_rows(
- get_content(file)
- ),
- v),
- files[v])),
- val))
+ ('{:>' + str(13 + name_column) + 's}').format(
+ 'Suma WK = {:.1f}'.format(wk_sum)))
+ return output
+
+
+ def format_histories(self, rows):
+ header = rows.pop(0)[0]
+ rows.pop(0)
+ if ' pauza ' in header:
+ return []
+ output = ['WYNIKI PARY NR ' + header,
+ '']
+ content_rows = []
+ add_separator = False
+ for row in rows:
+ content = []
+ if len(row) == 11:
+ add_separator = (
+ len(''.join(row[0:9])) == 0) and (
+ (add_separator is False) or (row[-2] == 'miejsce'))
+ content = row[0:4] + [
+ row[4] + ' ' + row[5] + ' ' + row[7]
+ ] + [row[6]] + row[8:11]
+ elif len(row) == 10:
+ content = [''] + row[0:3] + [
+ row[3] + ' ' + row[4] + ' ' + row[6]
+ ] + [row[5]] + row[7:10]
+ elif len(row) == 9:
+ content = ['', ''] + row[0:2] + [
+ row[2] + ' ' + row[3] + ' ' + row[5]
+ ] + [row[4]] + row[6:9]
+ if content:
+ if add_separator:
+ content_rows.append(
+ ['', '', '', '', '', '', '', '-------', '--------'])
+ content_rows.append(content)
+ else:
+ print 'histories: unexpected row length'
+ print row
+ column_width = max([len(r[1]) for r in content_rows])
+ content_rows = [[
+ 'RND', 'PRZECIWNIK', 'RZD', ' ', 'KONTRAKT', 'WIST',
+ 'ZAPIS', 'WYNIK ', u'/ BIEŻĄCY'
+ ]] + content_rows
+ for content in content_rows:
+ if content[6]:
+ score_align = u'>' if content[6][0] == u'-' else (
+ u'' if content[6][0] == u'+' else u'^')
+ else:
+ score_align = u''
+ output.append(
+ (u'{:>3s} {:' + unicode(column_width) +
+ u's} {:>3s} {:2s} {:9s}{:^4s} {:' +
+ score_align + u'7s} {:>7s}{:>8s}').format(
+ *[c or ' ' for c in content]))
+ output.insert(3, '-' * len(output[2]))
+ output.append('')
+ return output
+
+
+ def format_rows(self, rows, rowtype):
+ return getattr(self, 'format_' + rowtype)(rows)
+
+
+ def get_rows(self, content):
+ soup = BeautifulSoup(content, 'lxml')
+ output = []
+ link_regex = re.compile(r'^http://www\.msc\.com\.pl')
+ header = soup.find('h2')
+ if header:
+ output.append([header.text])
+ for table_row in soup.find_all('tr'):
+ row = map(lambda t:
+ os.linesep.join(t.stripped_strings),
+ table_row.find_all('td'))
+ row = row + map(lambda l:
+ l['href'],
+ table_row.find_all('a', {'href': link_regex}))
+ output.append(row)
+ return output
+
+
+ def get_content(self, filepath):
+ return re.sub('<img src=".*/(.*).gif" ?/>',
+ lambda img: img.group(1)[0].capitalize(),
+ open(filepath, 'r').read())
+
+
+ def get_header(self, directory):
+ soup = BeautifulSoup(
+ open(os.path.join(directory, 'index.html'), 'r').read(), 'lxml')
+ return [node.text for node in soup.select('#header *')]
+
+
+ def get_files(self, directory):
+ return dict(map(lambda (key, val): (
+ key,
+ reduce(list.__add__, map(
+ lambda v: sorted(glob(os.path.join(directory, v))), val), [])),
+ {
+ 'boards': ['d?.txt', 'd??.txt'],
+ 'protocols': ['p?.txt', 'p??.txt'],
+ 'histories': ['h?.txt', 'h??.txt'],
+ 'results': ['pary.txt'],
+ }.items()))
+
+
+ def compile_dir(self, directory):
+ files = self.get_files(directory)
+ return dict(
+ map(lambda (key, val):
+ (
+ key,
+ list(
+ chain(
+ *list(
+ i.next() for i in cycle(
+ map(lambda v:
+ iter(
+ map(lambda file:
+ self.format_rows(
+ self.get_rows(
+ self.get_content(file)
+ ),
+ v),
+ files[v])),
+ val))
+ )
)
)
- )
- ),
- {
- 'P': ['boards', 'protocols'],
- 'H': ['histories'],
- 'W': ['results']
- }.items()))
-
-directory = sys.argv[1] if len(sys.argv) > 1 else os.path.abspath('.')
-header = get_header(directory) + ['']
-output = compile_dir(directory)
-file_prefix = os.path.basename(directory)
-
-for filepath, rows in output.iteritems():
- output_file = open(file_prefix + filepath + '.txt', 'w')
- for line in header:
- output_file.write(line.encode('windows-1250') + '\n')
- for row in rows:
- output_file.write(row.encode('windows-1250') + '\n')
+ ),
+ {
+ 'P': ['boards', 'protocols'],
+ 'H': ['histories'],
+ 'W': ['results']
+ }.items()))
+
+ def generate(self, directory):
+ header = self.get_header(directory) + ['']
+ output = self.compile_dir(directory)
+ file_prefix = os.path.dirname(directory).split('/')[-1]
+
+ for filepath, rows in output.iteritems():
+ output_file = open(file_prefix + filepath + '.txt', 'w')
+ for line in header:
+ output_file.write(line.encode('windows-1250') + '\n')
+ for row in rows:
+ output_file.write(row.encode('windows-1250') + '\n')
+
+
+directory = sys.argv[1] if len(sys.argv) > 1 else '.'
+pair_file = os.path.join(directory, 'pary.txt')
+
+rrb_pdf = RRBPdfFix(pair_file)
+rrb_pdf.fixpdf()
+
+rrb_gen = RRBTxtGen()
+rrb_gen.generate(directory)
|