# -*- Mode: Python -*-
# vi:si:et:sw=4:sts=4:ts=4

# show all possible offsets, in order of popularity, from a download of
# http://www.accuraterip.com/driveoffsets.htm

import sys

from bs4 import BeautifulSoup

if len(sys.argv) < 2:
    print("Usage: %s driveoffsets_file" % sys.argv[0], file=sys.stderr)
    raise SystemExit(1)

with open(sys.argv[1]) as f:
    doc = f.read()

soup = BeautifulSoup(doc, features='html.parser')

offsets = {}  # offset -> total count

# skip first two spurious elements
rows = soup.findAll('tr')[2:]
for row in rows:
    columns = row.findAll('td')
    if len(columns) == 4:
        first, second, third, fourth = columns
        name = first.find(text=True)
        offset = second.find(text=True)
        count = third.find(text=True)

        # only use numeric offsets
        try:
            int(offset)
        except ValueError:
            continue

        if offset not in offsets.keys():
            offsets[offset] = 0
        offsets[offset] += int(count)

# now sort offsets by count
counts = []
for offset, count in offsets.items():
    counts.append((count, offset))

counts.sort()
counts.reverse()

offsets = []
for count, offset in counts:
    offsets.append(offset)

# now format it for code inclusion
lines = []
line = 'OFFSETS = ("'

for offset in offsets:
    line += offset + ', '
    if len(line) > 60:
        line += '"'
        lines.append(line)
        line = '           "'

# get last line too, trimming the comma and adding the quote
if len(line) > 11:
    line = line[:-2] + '")'
    lines.append(line)

print('\n'.join(lines))