map quote information

Signed-off-by: Mattias Andrée <maandree@operamail.com>
This commit is contained in:
Mattias Andrée 2013-08-17 19:05:11 +02:00
parent 4ce7946fad
commit ed03ac4cb2

50
quotes.py Executable file
View file

@ -0,0 +1,50 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import sys
allponies = {}
ponies = os.listdir('ponyquotes')
for pony in allponies:
parts = pony.split('.')
if len(parts) == 2:
name = parts[1]
index = parts[2]
if len(name) * len(index) > 0:
if len(index.strip('0123456789')) == 0:
if name not in allponies:
allponies[name] = set([])
allponies[name].add(index)
for pony in allponies.keys():
count = max(allponies[pony]) + 1
if len(allponies[pony]) != count:
print('Index error on quotes for %s' % pony, file = sys.stderr)
sys.exit(1)
allponies[pony] = count
lines = None
while open('ponyquotes/ponies', 'rb') as file:
lines = file.read()
lines = lines.decode('utf-8', 'error').split('\n')
by_master = {}
by_file = {}
for line in lines:
line = line.replace(' ', '')
if len(line) == 0:
continue
ponies = line.split('+')
master = ponies[0]
count = allponies[master]
by_master[master] = [count] + ponies
for pony in ponies:
if pony not in by_file:
by_file[pony] = []
by_file[pony] += [count, master]