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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
"""
Migrate from model_mommy to model_bakery.
``python from_mommy_to_bakery.py --dry-run``
Please check your dependency files.
"""
import argparse
import os
import re
PACKAGE_NAME = r"\bmodel_mommy\b"
PACKAGE_NAME_PATTERN = re.compile(PACKAGE_NAME)
PACKAGE_RECIPES = r"\bmommy_recipes\b"
PACKAGE_RECIPES_PATTERN = re.compile(PACKAGE_RECIPES)
PACKAGE_LEGACY_MODULE = r"\bmommy\b"
PACKAGE_LEGACY_MODULE_PATTERN = re.compile(PACKAGE_LEGACY_MODULE)
LEGACY_AND_NEW = [
{"old": PACKAGE_NAME_PATTERN, "new": r"model_bakery"},
{"old": PACKAGE_RECIPES_PATTERN, "new": r"baker_recipes"},
{"old": PACKAGE_LEGACY_MODULE_PATTERN, "new": r"baker"},
]
EXCLUDE = [
"node_modules",
"venv",
".git",
"sql",
"docs",
"from_mommy_to_bakery.py",
"Pipfile",
"Pipfile.lock",
]
def _find_changes(content, pattern, new_value):
new_content, substitutions = re.subn(pattern, new_value, content)
return new_content, substitutions > 0
def _rename_recipe_file(to_be_renamed, dry_run):
if dry_run is True:
print("Will be renamed:")
for old_recipe_file in to_be_renamed:
root = old_recipe_file[: old_recipe_file.rfind("/")]
if dry_run is False:
os.rename(old_recipe_file, f"{root}/baker_recipes.py")
else:
print(old_recipe_file)
def _replace_legacy_terms(file_path, dry_run):
with open(file_path, encoding="utf-8") as f:
content = f.read()
changed = []
for patterns in LEGACY_AND_NEW:
old, new = patterns["old"], patterns["new"]
content, has_changed = _find_changes(content, old, new)
changed.append(has_changed)
if any(changed):
if dry_run is False:
with open(file_path, "w", encoding="utf-8") as f:
f.write(content)
else:
print(file_path)
def _sanitize_folder_or_file(folder_or_file):
folder_or_file = folder_or_file.strip()
if folder_or_file.endswith("/"):
# Remove trailing slash e.g.: '.tox/' -> '.tox'
folder_or_file = folder_or_file[:-1]
return folder_or_file
def check_files(dry_run):
with open(".gitignore") as f:
excluded_by_gitignore = [
_sanitize_folder_or_file(folder_or_file) for folder_or_file in f.readlines()
]
exclude = EXCLUDE[:]
exclude.extend(excluded_by_gitignore)
to_be_renamed = []
for root, dirs, files in os.walk(".", topdown=True):
dirs[:] = [directory for directory in dirs if directory not in exclude]
for file_ in files:
if file_ in exclude or not file_.endswith(".py"):
continue
file_path = f"{root}/{file_}"
if file_ == "mommy_recipes.py":
to_be_renamed.append(file_path)
_replace_legacy_terms(file_path, dry_run)
_rename_recipe_file(to_be_renamed, dry_run)
if __name__ == "__main__":
description = "Help you to migrate from model_mommy to model_bakery."
parser = argparse.ArgumentParser(description=description)
parser.add_argument(
"--dry-run",
dest="dry_run",
action="store_true",
help="See which files will be changed.",
)
args = parser.parse_args()
check_files(args.dry_run)
|
