aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@archive.org>2020-06-23 20:09:33 -0700
committerBryan Newbold <bnewbold@archive.org>2020-06-23 20:09:33 -0700
commit3d73e4e52f1522a0c097b5f96576c15736e08284 (patch)
tree93f4894855a18fefa4852a2ff21db24216ce2c9a
parenta9ff88d6a0da3276eec0270847225f8f8bce7e94 (diff)
downloadchocula-3d73e4e52f1522a0c097b5f96576c15736e08284.tar.gz
chocula-3d73e4e52f1522a0c097b5f96576c15736e08284.zip
make fmt
-rw-r--r--chocula/util.py73
1 files changed, 39 insertions, 34 deletions
diff --git a/chocula/util.py b/chocula/util.py
index bff93ec..a52e1ec 100644
--- a/chocula/util.py
+++ b/chocula/util.py
@@ -123,27 +123,29 @@ OTHER_PUBLISHERS = [
def parse_lang(s: str) -> Optional[str]:
if not s or s in ("Not applicable", "Multiple languages", "Unknown"):
return None
- s = s.strip().split(',')[0].split()[0]
+ s = s.strip().split(",")[0].split()[0]
try:
lang = pycountry.languages.lookup(s)
- if lang.alpha_3 in ('mul', 'mis'):
+ if lang.alpha_3 in ("mul", "mis"):
return None
return lang.alpha_2.lower()
except LookupError:
- #print(f"unknown lang: {s}", file=sys.stderr)
+ # print(f"unknown lang: {s}", file=sys.stderr)
return None
except AttributeError:
print(f"partial lang for s={s}: {lang}", file=sys.stderr)
return None
+
def test_parse_lang():
- assert parse_lang('') is None
- assert parse_lang('asdf blah') is None
- assert parse_lang('en') == 'en'
- assert parse_lang('EN') == 'en'
- assert parse_lang('ENG') == 'en'
- assert parse_lang('English') == 'en'
- assert parse_lang('Portuguese') == 'pt'
+ assert parse_lang("") is None
+ assert parse_lang("asdf blah") is None
+ assert parse_lang("en") == "en"
+ assert parse_lang("EN") == "en"
+ assert parse_lang("ENG") == "en"
+ assert parse_lang("English") == "en"
+ assert parse_lang("Portuguese") == "pt"
+
def parse_country(s: str) -> Optional[str]:
if not s or s in ("Unknown"):
@@ -151,11 +153,11 @@ def parse_country(s: str) -> Optional[str]:
s = s.strip()
if s.lower() in ("usa", "new york (state)", "washington (state)"):
- return 'us'
+ return "us"
if s.lower() in ("russia (federation)", "russia"):
- return 'ru'
+ return "ru"
if s == "Québec (Province)":
- s = 'Canada'
+ s = "Canada"
if s == "China (Republic : 1949- )":
return "tw"
if s == "Brunei":
@@ -163,16 +165,16 @@ def parse_country(s: str) -> Optional[str]:
if s.startswith("Congo "):
s = "Congo"
if s.lower() == "iran":
- return 'ir'
+ return "ir"
if s.lower() == "bermuda islands":
- return 'bm'
+ return "bm"
if s.lower() == "burma":
- s = 'myanmar'
+ s = "myanmar"
if s.lower() in ("korea (south)", "south korea"):
- return 'kr'
+ return "kr"
if s.lower() in ("england", "scotland", "wales"):
- return 'uk'
- s = s.replace(' (Republic)', '').replace(" (Federation)", '')
+ return "uk"
+ s = s.replace(" (Republic)", "").replace(" (Federation)", "")
try:
country = pycountry.countries.lookup(s)
@@ -186,24 +188,25 @@ def parse_country(s: str) -> Optional[str]:
except LookupError:
sub = None
- s = s.replace(' (State)', '').replace(" (Province)", '')
+ s = s.replace(" (State)", "").replace(" (Province)", "")
if sub:
return sub.country_code.lower()
else:
- #print(f"unknown country: {s}", file=sys.stderr)
+ # print(f"unknown country: {s}", file=sys.stderr)
return None
+
def test_parse_country():
- assert parse_country('') is None
- assert parse_country('asdf blah') is None
- assert parse_country('us') == 'us'
- assert parse_country('USA') == 'us'
- assert parse_country('United States of America') == 'us'
- assert parse_country('united States') == 'us'
- assert parse_country('Massachusetts') == 'us'
- assert parse_country('Russia') == 'ru'
- assert parse_country('Japan') == 'jp'
+ assert parse_country("") is None
+ assert parse_country("asdf blah") is None
+ assert parse_country("us") == "us"
+ assert parse_country("USA") == "us"
+ assert parse_country("United States of America") == "us"
+ assert parse_country("united States") == "us"
+ assert parse_country("Massachusetts") == "us"
+ assert parse_country("Russia") == "ru"
+ assert parse_country("Japan") == "jp"
def parse_mimetypes(val: str) -> Optional[List[str]]:
@@ -219,11 +222,13 @@ def parse_mimetypes(val: str) -> Optional[List[str]]:
return None
return [mimetype]
+
def test_parse_mimetypes():
- assert parse_mimetypes('') is None
- assert parse_mimetypes('asdf blah') is None
- assert parse_mimetypes('application/pdf') == ['application/pdf']
- assert parse_mimetypes('PDF') == ['application/pdf']
+ assert parse_mimetypes("") is None
+ assert parse_mimetypes("asdf blah") is None
+ assert parse_mimetypes("application/pdf") == ["application/pdf"]
+ assert parse_mimetypes("PDF") == ["application/pdf"]
+
def gaps_to_spans(first, last, gaps):
if not gaps: