Rim/IW4MAdminDBParser
Rim/IW4MAdminDBParser
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Better config file handling

Browse Source
This commit is contained in:
Ahrimdon 2024-03-03 04:42:35 -05:00
parent ae9cd5005b
commit a8bb459638
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
parse_db.py
View File

@ -18,7 +18,7 @@ def setup_argparser():
parser.add_argument('-p', '--path', type=str, default="Database.db", help="Path to the IW4MAdmin database file.") parser.add_argument('-p', '--path', type=str, default="Database.db", help="Path to the IW4MAdmin database file.")
parser.add_argument('-o', '--output', type=str, help="Output directory for the parsed database file.") parser.add_argument('-o', '--output', type=str, help="Output directory for the parsed database file.")
parser.add_argument('-t', '--time', action='store_true', help="Time the script's execution.") parser.add_argument('-t', '--time', action='store_true', help="Time the script's execution.")
parser.add_argument('-c', '--config', type=str, help="Path to the configuration file.") parser.add_argument('-c', '--config', action='store_true', help="Use default config.json for configuration.")
return parser.parse_args() return parser.parse_args()
def delete_existing_db(output_db_path): def delete_existing_db(output_db_path):
@ -27,10 +27,11 @@ def delete_existing_db(output_db_path):
os.remove(output_db_path) os.remove(output_db_path)
print(f"Existing parsed database at {output_db_path} has been removed.") print(f"Existing parsed database at {output_db_path} has been removed.")
def load_or_create_config(config_path): def load_or_create_config():
config_path = "config.json" # Always use config.json
default_config = { default_config = {
"tables": { "tables": {
"AuditLog": 1000, # Example default value "AuditLog": 1000,
"ClientConnectionHistory": 1000, "ClientConnectionHistory": 1000,
"ClientMessages": 1000, "ClientMessages": 1000,
"Clients": 1000, "Clients": 1000,
@ -40,7 +41,7 @@ def load_or_create_config(config_path):
"Metadata": 1000, "Metadata": 1000,
"Penalties": 1000, "Penalties": 1000,
"PenaltyIdentifiers": 1000, "PenaltyIdentifiers": 1000,
"Servers": 1000 "Servers": 1000,
} }
} }
if not os.path.isfile(config_path): if not os.path.isfile(config_path):
@ -48,8 +49,9 @@ def load_or_create_config(config_path):
json.dump(default_config, config_file, indent=4) json.dump(default_config, config_file, indent=4)
print(f"Default configuration file created at {config_path}") print(f"Default configuration file created at {config_path}")
else: else:
with open(config_path, 'r') as config_file: with open(config_path) as config_file:
return json.load(config_file) return json.load(config_file)
return default_config
def main(): def main():
args = setup_argparser() args = setup_argparser()
@ -59,7 +61,7 @@ def main():
config = {} config = {}
if args.config: if args.config:
config = load_or_create_config(args.config) config = load_or_create_config()
db_path = args.path db_path = args.path
output_dir = args.output if args.output else '.' # Use current directory if no output directory is specified output_dir = args.output if args.output else '.' # Use current directory if no output directory is specified