1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#!/usr/bin/env python3
# coding: utf-8
# update/nz_banks.py - script to download Bank list from Bank Branch Register
#
# Copyright (C) 2019-2021 Arthur de Jong
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301 USA
"""This script downloads the list of banks with bank codes as used in the
New Zealand bank account numbers."""
import re
from collections import OrderedDict, defaultdict
import requests
import xlrd
# Monkey patch xlrd avoiding bug in combination with Python 3.9
xlrd.xlsx.ensure_elementtree_imported(False, None)
xlrd.xlsx.Element_has_iter = True
# The page that contains a link to the latest XLS version of the codes.
download_url = 'https://www.paymentsnz.co.nz/resources/industry-registers/bank-branch-register/download/xlsx/'
def get_values(sheet):
"""Return rows from the worksheet as a dict per row."""
rows = sheet.get_rows()
# the first row has column names
columns = [column.value.lower().replace(' ', '_') for column in next(rows)]
# go over rows with values
for row in rows:
yield dict(zip(columns, [column.value for column in row]))
def branch_list(branches):
"""Return a compact representation of a list of branch numbers."""
branches = sorted(int(b) for b in branches)
first = None
prev = None
res = ''
for branch in branches:
if first is not None and branch == prev + 1:
# this branch is consecutive to the previous: make a range
if prev > first:
res = res[:-5]
res += '-%04d' % branch
prev = branch
else:
# this is a new branch, add a new one to the list
res += ',%04d' % branch
first = prev = branch
return res.lstrip(',')
if __name__ == '__main__':
# parse the download as an XLS
response = requests.get(download_url, timeout=30)
response.raise_for_status()
content_disposition = response.headers.get('content-disposition', '')
filename = re.findall(r'filename=?(.+)"?', content_disposition)[0].strip('"')
workbook = xlrd.open_workbook(file_contents=response.content)
sheet = workbook.sheet_by_index(0)
# print header
print('# generated from %s downloaded from' % filename)
print('# %s' % download_url)
# build banks list from spreadsheet
banks = defaultdict(dict)
for line in get_values(sheet):
banks[line['bank_number']]['bank'] = line['bank_name']
branches = banks[line['bank_number']].setdefault('branches', OrderedDict())
branches.setdefault((line['branch_information'], line['bic']), list()).append(line['branch_number'])
# output bank information
for bank_number in sorted(banks.keys()):
bank = banks[bank_number]
print('%s bank="%s"' % (bank_number, bank['bank']))
for (branch, bic), branch_numbers in bank['branches'].items():
print(' %s%s branch="%s"' % (
branch_list(branch_numbers), ' bic="%s"' % bic if bic else '', branch))
|