-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
56 lines (38 loc) · 1.17 KB
/
main.py
File metadata and controls
56 lines (38 loc) · 1.17 KB
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
import logging
import sys
from pathlib import Path
from urllib.parse import urlparse
import click
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
@click.command()
@click.argument("problem")
def cli(problem: str) -> None:
slug = parse_slug(problem)
create_templates(slug)
def exists(path: Path) -> None:
if path.exists():
logger.error("%s already exists", path)
sys.exit(1)
def parse_slug(problem: str) -> str:
if problem.startswith("https"):
url = urlparse(problem)
segments = url.path.split("/")
assert not segments[0] and segments[1] == "problems", f"invalid {url=}"
return segments[2]
else:
return problem
def create_templates(name: str) -> None:
filename = name.replace("-", "_")
if filename[0].isnumeric():
filename = f"_{filename}"
src_file = Path("src", f"{filename}.py")
test_file = Path("tests", f"test_{filename}.py")
exists(src_file)
exists(test_file)
src_file.open("w").close()
test_file.open("w").close()
logger.info("touch %s", src_file)
logger.info("touch %s", test_file)
if __name__ == "__main__":
cli()