Skip to content

Commit

Permalink
#259 review fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
Tim Van Campenhout committed Nov 21, 2024
1 parent 722f449 commit e179bc6
Showing 1 changed file with 16 additions and 4 deletions.
20 changes: 16 additions & 4 deletions crabpy/gateway/adressenregister.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,7 +243,19 @@ def get_provincie_by_niscode(self, niscode):
return provincie
return None

def list_gemeenten_by_provincie(self, provincie):
def gemeente_by_status(self, status):
"""
Filter gemeenten by status.
:param status: The status for which the `gemeenten` are wanted.
:rtype: A :class:`list` of :class:`Gemeente`.
"""
return (
[gemeente for gemeente in self.gemeenten if gemeente.status == status]
if status else self.gemeenten
)

def list_gemeenten_by_provincie(self, provincie, gemeente_status=None):
"""
List all `gemeenten` in a `provincie`.
Expand All @@ -258,11 +270,11 @@ def list_gemeenten_by_provincie(self, provincie):
provincie_niscode = provincie.niscode
return [
gemeente
for gemeente in self.gemeenten
for gemeente in self.gemeente_by_status(gemeente_status)
if gemeente.provincie_niscode == provincie_niscode
]

def list_gemeenten(self, gewest_niscode="2000"):
def list_gemeenten(self, gewest_niscode="2000", gemeente_status=None):
"""
List all `gemeenten` in a `gewest`.
Expand All @@ -274,7 +286,7 @@ def list_gemeenten(self, gewest_niscode="2000"):
if gewest_niscode == "4000":
return [
gemeente
for gemeente in self.gemeenten
for gemeente in self.gemeente_by_status(gemeente_status)
if gemeente.niscode.startswith("21")
]
provincie_niscodes = [
Expand Down

0 comments on commit e179bc6

Please sign in to comment.