def test_writer_one_list_item():
out = io.StringIO()
writer = Writer(out, fieldnames=["name"])
item = Item(name=["one", "two", "three"])
writer.write(item)
writer.close()
string = out.getvalue()
assert string == 'name\none;two;three\n'
def test_writer_one_item():
out = io.StringIO()
writer = Writer(out, fieldnames=["name"])
item = Item(name="one")
writer.write(item)
writer.close()
string = out.getvalue()
assert string == 'name\none\n'
def test_writer_sparse_items():
out = io.StringIO()
writer = Writer(out, fieldnames=["name", "text", "address", "z"])
for name in ['one', 'two', 'three']:
item = Item(name=name, text="hello world")
writer.write(item)
writer.close()
string = out.getvalue()
assert string == ('name\ttext\taddress\tz\n'
'one\thello world\t\t\n'
'two\thello world\t\t\n'
'three\thello world\t\t\n')
def test_writer_many_items():
out = io.StringIO()
writer = Writer(out, fieldnames=["name", "text"])
for name in ["one", "two", "three"]:
item = Item(name=name, text="hello world")
writer.write(item)
writer.close()
string = out.getvalue()
assert string == ("name\ttext\n"
"one\thello world\n"
"two\thello world\n"
"three\thello world\n")
#!/usr/bin/env python3
import sys
import json
from openregister import Item
from openregister.representations.tsv import Writer
fields = ['police-force', 'name', 'start-date', 'end-date']
writer = Writer(sys.stdout, fieldnames=fields)
for row in json.load(sys.stdin):
force = Item()
force['police-force'] = row['id']
force['name'] = row['name']
writer.write(force)
writer.close()
# Country,Competent Authority,Approval Number,Trading name,Address 1,Address 2,Adress 3,Town/Region,Postcode,Section 0,Section I,Section II,Section III,Section IV,Section V,Section VI,Section VII,Section VIII,Section IX,Section X,Section XI,Section XII,Section XIII,Section XIV,Section XV,Section XVI,Species,Remarks,Address withheld,uprn,lat,lng,match_score,match_description
fields = ["food-premises", "premises", "name", "business", "start-date", "end-date"]
writer = Writer(sys.stdout, fields)
reader = csv.DictReader(sys.stdin)
for num, row in enumerate(reader):
uprn = row["uprn"]
name = row["Trading name"]
if uprn and row["Address withheld"] != "Y":
item = Item()
item.__dict__["food-premises"] = uprn
if uprn in uprns:
item.premises = uprns[uprn]
if name in companies:
if companies[name]:
item.business = "company:" + companies[name]
item.__dict__["end-date"] = ""
item.name = row["Trading name"]
writer.write(item)
writer.close()
