一套小工具
本文作者 Yile Wang，在 GNU Free Documentation License version 1.3 下发布。不提供任何担保。
本文给出的部分代码在 GNU General Public License version 3, or any later version 下发布，详见对应代码的文件头声明。
COPYING
Copyright (c) 2024 Yile Wang
Permission is granted to copy, distribute and/or modify this document
under the terms of the GNU Free Documentation License, Version 1.3
or any later version published by the Free Software Foundation;
with no Invariant Sections, no Front-Cover Texts, and no Back-Cover Texts.
A copy of the license is included in the section entitled "GNU
Free Documentation License".
BRIEFING
这是一个小工具集合，作者希望它有用，但不提供任何担保。
MENU
- random_prime
- checker
- nospace
- sweeper
- find_problem
random_prime
在 \([l, r]\) 之间随机选择一个数，找出第一个不小于它的质数并输出。
import math
def chkprime(x: int) -> bool:
p = int(math.sqrt(x))
if x == 1:
return False
if x == 2:
return True
for i in range(2, p + 1):
if x % i == 0:
return False
return True
import random
l = int(1e9)
r = int(2e9)
x = random.randint(l, r)
while not chkprime(x):
x += 1
print(x)
checker
简单的文件比对，推荐更强大的比对工具 GNU Diffutils。
out = "/path/to/out"
ans = "/path/to/ans"
with open(out) as file:
out = file.readlines()
with open(ans) as file:
ans = file.readlines()
for i in range(len(ans)):
if len(out) <= i:
print("miss")
break
if out[i] != ans[i]:
print("err:", i + 1)
nospace
去除 C++ 代码文件中多余的空格，你可以通过调整代码开头的选项改变其行为。对代码的修改也应使用 GNU 通用公共许可证（第三版以上）发布。
"""
nospace.py
Copyright (C) 2024 Yile Wang
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
======================================================================
You can contact the author by email. (mailto:bluewindde@163.com)
This program expects to do as follow: remove unnecessary space from
C++ source file.
"""
FIRST_OF_ALL = """
nospace.py Copyright (C) 2024 Yile Wang
This program comes with ABSOLUTELY NO WARRANTY.
This is free software, and you are welcome to redistribute it
under certain conditions.
Learn more in the copyright statement at the beginning
of the source code.
"""
print(FIRST_OF_ALL)
# 忽略注释 NotImplemented
ignore_comment = False
# 忽略行尾空格
ignore_ending_space = False
# 忽略预编译指令
ignore_precopile_command = True
path = "/path/to/cpp"
with open(path, "r", encoding="utf-8") as file:
text = file.read().splitlines()
def chk(x: str, y: str) -> bool:
flag1 = flag2 = False
if x.islower() or x.isupper() or x.isdigit():
flag1 = True
if y.islower() or y.isupper() or y.isdigit():
flag2 = True
return (flag1 and flag2) or x == " " or y == " "
ret = []
for i in range(len(text)):
if ignore_ending_space:
para = text[i]
else:
para = text[i].rstrip()
if ignore_precopile_command and len(para) != 0 and para[0] == "#":
ret.append(para)
continue
ret.append("")
for j in range(len(para)):
ch = para[j]
if ch == " ":
if j > 0 and j < len(para) - 1 and not chk(para[j - 1], para[j + 1]):
continue
ret[-1] += ch
with open(path + ".ret.cpp", "w", encoding="utf-8") as file:
for para in ret:
file.write(para)
file.write("\n")
sweeper
寻找当前工作目录下可能无用的文件，并询问是否删除。
你可以通过修改源代码自定义程序的行为，部分参数在代码中有注释说明。请注意，你修改后的程序仍应以 GNU 通用公共许可证（第三版以上）发布。
"""
sweeper.py
Copyright (C) 2024 Yile Wang
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
======================================================================
You can contact the author by email. (mailto:bluewindde@163.com)
This program expects to do as follow: walk current directory then find
maybe-useless files.
"""
FIRST_OF_ALL = """
sweeper.py Copyright (C) 2024 Yile Wang
This program comes with ABSOLUTELY NO WARRANTY.
This is free software, and you are welcome to redistribute it
under certain conditions.
Learn more in the copyright statement at the beginning
of the source code.
"""
print(FIRST_OF_ALL)
import os
def isexec(path: str):
"""
如果 path 指向 ELF 文件，返回 True；否则返回 False
"""
try:
file = open(path, "rb")
if file.read(4)[1:] == b"ELF":
return True
return False
except OSError:
return False
ignore = [] # 不扫描的子文件夹，相对路径和绝对路径均可
def isignore(path: str) -> bool:
"""
如果 path 指向被忽略的文件夹，返回 True；否则返回 False
"""
path = os.path.normpath(path)
for dir in ignore:
if dir.startswith("."):
dir = os.path.join(os.getcwd(), dir)
dir = os.path.normpath(dir)
if path == dir:
print(f"\033[34mignore:\033[0m {path}")
return True
elif path.startswith(dir):
return True
exts = [] # 被认为“无用”的文件拓展名
def iswaste(path: str) -> bool:
"""
如果 path 指向无用文件，返回 True；否则返回 False
"""
for ext in exts:
if path.endswith(ext):
return True
if isexec(path):
return True
return False
def sweep(path: str):
"""
清扫 path 指向的文件夹
"""
aim = []
for dirpath, dirnames, filenames in os.walk(path):
if isignore(dirpath):
continue
print(f"\033[34msweeping:\033[0m {dirpath}:")
for file in filenames:
print(f"\tchecking: {file}...", end="")
path = os.path.join(dirpath, file)
flag = iswaste(path)
if flag == True:
print("\033[33mwaste.\033[0m")
aim.append(path)
elif flag is None:
print("\033[31mpermission denied.\033[0m")
else:
print("\033[32mok.\033[0m")
if len(aim) == 0:
print("won't remove any file.")
return
elif len(aim) == 1:
print("would remove 1 file:")
else:
print(f"would remove {len(aim)} files:")
for file in aim:
print(f"\t{file}")
confirm = input("confirm ? [Y/n] ").strip()
if confirm == "Y" or confirm == "y":
failed = []
for file in aim:
try:
os.remove(file)
except Exception as msg:
failed.append((file, msg))
if len(failed) > 0:
print("\033[31mfailed\033[0m while removing files below:")
for file, err in failed:
print(f"\t{file}: {err}")
else:
print("\033[32msuccess.\033[0m")
if __name__ == "__main__":
try:
os.chdir(os.path.dirname(__file__))
except OSError:
pass
sweep(os.getcwd())
find_problem
尝试寻找可能为算法竞赛题目编号的字段并输出。
"""
find_problem.py
Copyright (C) 2024 Yile Wang
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
======================================================================
You can contact the author by email. (mailto:bluewindde@163.com)
This program expects to do as follow: read a Markdown file,
find and output possible problem from competitive programming.
"""
FIRST_OF_ALL = """
find_problem.py Copyright (C) 2024 Yile Wang
This program comes with ABSOLUTELY NO WARRANTY.
This is free software, and you are welcome to redistribute it
under certain conditions.
Learn more in the copyright statement at the beginning
of the source code.
"""
print(FIRST_OF_ALL)
path = input("Type or paste the path to file here: ")
import re
patterns = [
r"\bLuogu [BP][0-9]+\b",
r"\bCodeForces [1-9][0-9]*[A-Z][1-9]?\b",
r"\b((AGC)|(ABC)|(ARC))[0-9]+[A-Z]\b",
r"\bSPOJ [A-Z]+[2-9]?\b",
r"\bUVA[1-9][0-9]*\b",
r"\bUOJ[1-9][0-9]*\b",
r"\bLOJ[1-9][0-9]*\b",
]
try:
ret = []
with open(path, "r", encoding="utf-8") as file:
ln = 0
for line in file.readlines():
ln += 1
if line.strip().startswith("-") or line.strip().startswith(">"):
continue
for pattern in patterns:
cur = re.findall(pattern, line)
ret += cur
for id in cur:
print(f"ln {ln}: {id}")
except Exception as err:
print(f"error: {err}")
else:
print()
print(f"In total: {len(ret)}")
print("Problem ID are as follow:")
for id in ret:
print(id, end=", ")
print()