Skip to content

Composite pattern applied to countries #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
anjilag wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Composite pattern applied to countries #2

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 46 additions & 11 deletions app/utils/countries.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
"""app.utils.countries.py"""
import logging
from abc import ABCMeta, abstractmethod

LOGGER = logging.getLogger(__name__)

Expand Down Expand Up @@ -366,15 +367,49 @@
# "MS Zaandam"
}

# fmt: on
def country_code(value):
"""
Return two letter country code (Alpha-2) according to https://en.wikipedia.org/wiki/ISO_3166-1
Defaults to "XX".
"""
code = COUNTRY_NAME__COUNTRY_CODE.get(value, DEFAULT_COUNTRY_CODE)
if code == DEFAULT_COUNTRY_CODE:
# log at sub DEBUG level
LOGGER.log(5, f"No country code found for '{value}'. Using '{code}'!")
class CountriesComponent(metaclass=ABCMeta):
"""Interface"""
@abstractmethod
def country_code(self): pass
def country_name(self): pass

return code
class CountriesGroup(CountriesComponent):
"""The Composite: A list of countries grouped in some manner"""
countriesList = []

def __init__(self, countriesList):
self.countriesList = countriesList

def country_code(self):
codeList = []
for country in self.countriesList:
codeList.append(country.country_code())
return codeList

def country_name(self):
nameList = []
for country in self.countriesList:
nameList.append(country.country_code())
return nameList

class Country(CountriesComponent):
"""The Leaf: A singular country"""
def __init__(self, name, code):
self.name = name
self.code = code

def country_name(self):
return self.name

# fmt: on
def country_code(value):
"""
Return two letter country code (Alpha-2) according to https://en.wikipedia.org/wiki/ISO_3166-1
Defaults to "XX".
"""
code = COUNTRY_NAME__COUNTRY_CODE.get(value, DEFAULT_COUNTRY_CODE)
if code == DEFAULT_COUNTRY_CODE:
# log at sub DEBUG level
LOGGER.log(5, f"No country code found for '{value}'. Using '{code}'!")

return code