More robust env.sh | Fix conda activate if in subshell
This commit is contained in:
56
scripts/replace-content.py
Executable file
56
scripts/replace-content.py
Executable file
@@ -0,0 +1,56 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
# Replace text in a file given a key identifying a block to replace.
|
||||
# If the file doesn't exist, create it.
|
||||
#
|
||||
# args
|
||||
# $1 - file to replace text in
|
||||
# $2 - key used to find block of text to replace
|
||||
# $3 - text to fill in block that is replaced
|
||||
|
||||
import re
|
||||
import sys
|
||||
|
||||
def CY_INITIALIZE_RE_BLOCK(k):
|
||||
return (
|
||||
r"^# >>> " + f"{k}" + r" initialize >>>(?:\n|\r\n)"
|
||||
r"([\s\S]*?)"
|
||||
r"# <<< " + f"{k}" + r" initialize <<<(?:\n|\r\n)?"
|
||||
)
|
||||
|
||||
def CY_INITIALIZE_START_TOKEN(k):
|
||||
return "# >>> " + f"{k}" + " initialize >>>"
|
||||
|
||||
def CY_INITIALIZE_END_TOKEN(k):
|
||||
return "# <<< " + f"{k}" + " initialize <<<"
|
||||
|
||||
# ------------------------------
|
||||
|
||||
try:
|
||||
with open(sys.argv[1]) as fh:
|
||||
fh_content = fh.read()
|
||||
except FileNotFoundError:
|
||||
fh_content = ""
|
||||
except:
|
||||
raise
|
||||
|
||||
initialize_comment_key = sys.argv[2]
|
||||
inner_contents = CY_INITIALIZE_START_TOKEN(initialize_comment_key) + "\n" + sys.argv[3] + "\n" + CY_INITIALIZE_END_TOKEN(initialize_comment_key) + "\n"
|
||||
|
||||
# ------------------------------
|
||||
|
||||
replace_str = "__CY_REPLACE_ME_123__"
|
||||
fh_content = re.sub(
|
||||
CY_INITIALIZE_RE_BLOCK(initialize_comment_key),
|
||||
replace_str,
|
||||
fh_content,
|
||||
flags=re.MULTILINE,
|
||||
)
|
||||
# TODO: maybe remove all but last of replace_str, if there's more than one occurrence
|
||||
fh_content = fh_content.replace(replace_str, inner_contents)
|
||||
|
||||
if CY_INITIALIZE_START_TOKEN(initialize_comment_key) not in fh_content:
|
||||
fh_content += "\n%s\n" % inner_contents
|
||||
|
||||
with open(sys.argv[1], "w") as fh:
|
||||
fh.write(fh_content)
|
||||
Reference in New Issue
Block a user