kfm/to-hen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve type checking: Make Url newtype

Browse Source
This commit is contained in:
kmein
2018-05-17 10:02:24 +02:00
parent 8db79b6c48
commit d967b8c44e
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
rilke.py
View File

@@ -3,25 +3,25 @@ from enum import Enum
from bs4 import BeautifulSoup
import re
import requests
from typing import Iterator
from typing import Iterator, NewType
class Language(Enum):
EN = "en"
DE = "de"
DEFAULT_LANG = Language.DE
Url = str
Url = NewType("Url", str)
def cook_soup(url: Url) -> BeautifulSoup:
return BeautifulSoup(requests.get(url).text, "lxml")
def base_url(lang: Language = DEFAULT_LANG) -> str:
return "https://{}.wiktionary.org".format(lang.value)
def base_url(lang: Language = DEFAULT_LANG) -> Url:
return Url("https://{}.wiktionary.org".format(lang.value))
def entry_url(word: Url, lang: Language = DEFAULT_LANG) -> str:
return base_url(lang) + "/wiki/" + word
def entry_url(word: str, lang: Language = DEFAULT_LANG) -> Url:
return Url(base_url(lang) + "/wiki/" + word)
def rhymes_url(entry_url: Url, lang: Language = DEFAULT_LANG) -> str:
def rhymes_url(entry_url: Url, lang: Language = DEFAULT_LANG) -> Url:
soup = cook_soup(entry_url)
result_url = base_url(lang)
try: