diff options
Diffstat (limited to 'srb.py')
| -rwxr-xr-x | srb.py | 85 |
1 files changed, 85 insertions, 0 deletions
@@ -0,0 +1,85 @@ +#!/usr/bin/env python3 +# Startdate: 2024-03-08-6 15:28 +# File: srb_lib.py +# Purpose: frontend for srb_lib +# History: +# Usage: +# Reference: +# blog posts 2024-03 +# delsum from github +# bgconf.py + +# WORKHERE: right now, this only just corrects any file. in the future, offer options like --profile 1 --set-money 3045 --give-plane all or --give-plane marcie +# or --take-plan marcie or --set-rank-level 1,general --set-rank-level 1,none --set-rank-level all,general +# or --give-weapon pumpkin --take-weapon bees +# or --give-balloons-level 10 --take-balloons-level 11 +# make chart of level numbers, which ones have balloons +# or --reset-level 5 (which zeros out all info about completion of that level) +# --set-plane-health 1 (through 4) +# --set-plane-weapon 1 (through 4) +# --set-plane-stunt 1 (through 4) +# --set-profile-name "asdbdf" + +import srb_lib, argparse, sys +from srb_lib import ferror, debuglev + +parser = argparse.ArgumentParser(description="Cli tool for manipulating savegame files for srb.exe") +parser.add_argument("-V","--version",action="version",version="%(prog)s " + srb_lib.srb_lib_version) +parser.add_argument("-d","--debug",nargs='?',default=0,type=int,choices=range(0,11),help="Set debug level") +parser.add_argument("--profile",type=int,choices=range(1,4),help="Profile in user menu.") +parser.add_argument("--get-money",action="store_true",help="Print current money for profile.") +parser.add_argument("--set-money",type=int,help="Set money for profile.") +parser.add_argument("--get-weapon",action="store_true",help="Print currently equipped weapon for profile.") +# choices seems to be too strict here for the numbers. We can live with just the +#choices=[i for i in srb_lib.WEAPONS if i != "undefined"]+list(range(0,16)) +#parser.add_argument("--set-weapon",choices=[i for i in srb_lib.WEAPONS if i != "undefined"],help="Print currently equipped weapon for profile.") +parser.add_argument("--set-weapon",help="Print currently equipped weapon for profile.") +parser.add_argument("--checksum",action=argparse.BooleanOptionalAction,default=True,help="Correct checksum. Default is to do this. It happens at the end of everything else.") +parser.add_argument("file",default="Profile 1.sav") +args = parser.parse_args() +debuglevel = 0 +if args.debug: + debuglevel = args.debug + +if debuglev(1,debuglevel): + ferror("debug level", debuglevel) +if debuglev(8,debuglevel): + ferror(args) + +# common parameters +profile_id = args.profile + +print(f"profile_id={profile_id}") + +if not profile_id and (args.get_money or args.set_money or args.get_weapon or args.set_weapon): + ferror("Warning: Cannot perform most actions without --profile. Not all tasks may run.") +else: + if args.get_money: + money = srb_lib.get_money(args.file, profile_id) + print(f"Profile {profile_id} has {money} money.") + if args.set_money: + if args.set_money > 0xFFFF: + ferror(f"Warning: Do not set money higher than 65535. While it can technically work, there's no need for that in-game anyways. Continuing...") + else: + data = srb_lib.set_money(args.file, profile_id, args.set_money) + srb_lib.write_file(args.file,0,data) + if args.get_weapon: + print(f"Profile {profile_id} has weapon {srb_lib.get_weapon(args.file, profile_id)}") + if args.set_weapon: + try: + args.set_weapon = int(args.set_weapon) + except: + pass + data = srb_lib.set_weapon(args.file, profile_id, args.set_weapon) + if type(data) == int and data == -1: + # error is printed in the function + pass + else: + srb_lib.write_file(args.file,0,data) + +if args.checksum: + f = args.file + #for f in args.file: + if debuglev(1,debuglevel): + ferror(f"Fixing checksum for file {f}") + srb_lib.correct_file(f,debuglevel) |