MVP register client

gistfile1.txt

import io
import csv
import requests


class Records:
    "Minimal register client"

    records = {}
    url = 'https://%s.register.gov.uk/records.tsv?page-size=5000'

    def load(self, register):
        url = self.url % (register)
        resp = requests.get(url=url)
        resp.raise_for_status()

        for row in csv.DictReader(io.StringIO(resp.text), delimiter='\t'):
            self.records['%s:%s' % (register, row[register])] = row

    def dump(self, fields, sep='\t'):
        key = fields[0]
        print(sep.join(fields))
        for record in sorted(self.records):
            row = self.records[record]
            row[key] = record

            print(sep.join([str(row[field]) for field in fields]))