72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
# Super Lead rope mod
|
|
# Copyright (C) 2025 R3944Realms
|
|
# 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/>.
|
|
import os
|
|
import random
|
|
|
|
# 确保输出目录存在
|
|
os.makedirs("dict", exist_ok=True)
|
|
|
|
# 全局已使用名字集合，保证三类字典互不重复
|
|
used_names = set()
|
|
|
|
def gen_dict(count: int, length: int, filename: str, charset: str):
|
|
"""
|
|
生成字典文件：
|
|
- 首字符必须是字母
|
|
- 剩余字符可以是 charset 中任意字符
|
|
- 保证全局唯一
|
|
"""
|
|
if length < 1:
|
|
raise ValueError("长度必须至少为 1")
|
|
|
|
# 首字符必须是字母
|
|
first_chars = [c for c in charset if c.isalpha()]
|
|
if not first_chars:
|
|
raise ValueError("字符集必须包含至少一个字母作为首字符")
|
|
|
|
chars = list(charset)
|
|
max_combinations = len(first_chars) * (len(chars) ** (length - 1))
|
|
if count > max_combinations:
|
|
raise ValueError(f"要求数量 {count} 超过最大可能组合数 {max_combinations}")
|
|
|
|
generated = set()
|
|
while len(generated) < count:
|
|
s = random.choice(first_chars)
|
|
s += ''.join(random.choice(chars) for _ in range(length - 1))
|
|
if s in used_names:
|
|
continue # 跳过已使用的名字
|
|
generated.add(s)
|
|
used_names.add(s)
|
|
|
|
with open(filename, "w") as f:
|
|
for s in generated:
|
|
f.write(s + "\n")
|
|
|
|
print(f"生成完成，共 {len(generated)} 个写入 {filename}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# 字符集定义
|
|
charset_class = "o0OQ"
|
|
charset_member = "i1lI"
|
|
charset_package = "UvuV"
|
|
|
|
# 生成类名字典
|
|
gen_dict(count=1000, length=10, filename="dict/class_dict.txt", charset=charset_class)
|
|
|
|
# 生成字段/方法字典
|
|
gen_dict(count=1000, length=10, filename="dict/member_dict.txt", charset=charset_member)
|
|
|
|
# 生成包名字典
|
|
gen_dict(count=1000, length=10, filename="dict/package_dict.txt", charset=charset_package)
|