1 Star 2 Fork 2

BlackTrigger / AV_Data_Capture

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
config.py 3.71 KB
一键复制 编辑 原始数据 按行查看 历史
68cdrBxM8YdoJ 提交于 2020-04-18 14:34 . Create config.py
import os
import configparser
class Config:
def __init__(self, path: str = "config.ini"):
if os.path.exists(path):
self.conf = configparser.ConfigParser()
self.conf.read(path, encoding="utf-8")
else:
print("[-] Config file not found! Use the default settings")
self.conf = self._default_config()
def main_mode(self) -> str:
try:
return self.conf.getint("common", "main_mode")
except ValueError:
self._exit("common:main_mode")
def failed_folder(self) -> str:
return self.conf.get("common", "failed_output_folder")
def success_folder(self) -> str:
return self.conf.get("common", "success_output_folder")
def soft_link(self) -> bool:
return self.conf.getboolean("common", "soft_link")
def proxy(self) -> [str, int, int]:
try:
sec = "proxy"
proxy = self.conf.get(sec, "proxy")
timeout = self.conf.getint(sec, "timeout")
retry = self.conf.getint(sec, "retry")
return proxy, timeout, retry
except ValueError:
self._exit("common")
def naming_rule(self) -> str:
return self.conf.get("Name_Rule", "naming_rule")
def location_rule(self) -> str:
return self.conf.get("Name_Rule", "location_rule")
def update_check(self) -> bool:
try:
return self.conf.getboolean("update", "update_check")
except ValueError:
self._exit("update:update_check")
def sources(self) -> str:
return self.conf.get("priority", "website")
def escape_literals(self) -> str:
return self.conf.get("escape", "literals")
def escape_folder(self) -> str:
return self.conf.get("escape", "folders")
def debug(self) -> bool:
return self.conf.getboolean("debug_mode", "switch")
@staticmethod
def _exit(sec: str) -> None:
print("[-] Read config error! Please check the {} section in config.ini", sec)
input("[-] Press ENTER key to exit.")
exit()
@staticmethod
def _default_config() -> configparser.ConfigParser:
conf = configparser.ConfigParser()
sec1 = "common"
conf.add_section(sec1)
conf.set(sec1, "main_mode", "1")
conf.set(sec1, "failed_output_folder", "failed")
conf.set(sec1, "success_output_folder", "JAV_output")
conf.set(sec1, "soft_link", "0")
sec2 = "proxy"
conf.add_section(sec2)
conf.set(sec2, "proxy", "127.0.0.1:1080")
conf.set(sec2, "timeout", "10")
conf.set(sec2, "retry", "3")
sec3 = "Name_Rule"
conf.add_section(sec3)
conf.set(sec3, "location_rule", "actor + '/' + number")
conf.set(sec3, "naming_rule", "number + '-' + title")
sec4 = "update"
conf.add_section(sec4)
conf.set(sec4, "update_check", "1")
sec5 = "priority"
conf.add_section(sec5)
conf.set(sec5, "website", "javbus,javdb,fanza,xcity,mgstage,fc2,avsox,jav321,xcity")
sec6 = "escape"
conf.add_section(sec6)
conf.set(sec6, "literals", "\()/") # noqa
conf.set(sec6, "folders", "failed, JAV_output")
sec7 = "debug_mode"
conf.add_section(sec7)
conf.set(sec7, "switch", "0")
return conf
if __name__ == "__main__":
config = Config()
print(config.main_mode())
print(config.failed_folder())
print(config.success_folder())
print(config.soft_link())
print(config.proxy())
print(config.naming_rule())
print(config.location_rule())
print(config.update_check())
print(config.sources())
print(config.escape_literals())
print(config.escape_folder())
print(config.debug())
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/soulbeanrie/AV_Data_Capture.git
git@gitee.com:soulbeanrie/AV_Data_Capture.git
soulbeanrie
AV_Data_Capture
AV_Data_Capture
master

搜索帮助

344bd9b3 5694891 D2dac590 5694891