177 lines
6.5 KiB
Python
177 lines
6.5 KiB
Python
import encode, decode, shared
|
|
import tables
|
|
import os
|
|
|
|
conf={}
|
|
confp=os.path.expanduser('~/.config/swat.cfg') # This won't work on Windows. You'll need to tweak this to run it on Windows. Not my problem.
|
|
|
|
|
|
defaults={'classes':[(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(0,0),(0,0),(1,3),(1,3),(1,3)],'guns':[(1,3),(1,3),(1,3),(1,3),(1,3),(0,0),(0,0),(1,3)],'armour':[(1,3),(1,3),(1,3),(1,3)],'traits':[(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3)],'specs':[(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3)],'talents':[(12,12),(1,3),(1,3),(1,3),(1,3),(1,3),(1,3)],'medals':{}}
|
|
def add_user(name):
|
|
namehash=encode.hash_name(name) # Store the hash so that if they request a different name that happens to have the same hash, it can just be stored together.
|
|
if namehash in conf: conf[namehash]['names'].add(name)
|
|
else: conf[namehash]=defaults.copy()|{'names':{name}}
|
|
save_conf()
|
|
global active_profile
|
|
active_profile=namehash
|
|
|
|
def parse_conf_line(line):
|
|
a=line.split(' ')
|
|
out=[]
|
|
for n in a:
|
|
b=n.split('/')
|
|
out.append((int(b[0]),int(b[1])))
|
|
return out
|
|
|
|
def load_conf():
|
|
if not os.path.exists(confp): return
|
|
a=open(confp).read()
|
|
for n in a.split('\n\n'):
|
|
lines=n.split('\n')
|
|
lines=list(filter(lambda x:x,lines))
|
|
if not lines: continue
|
|
header=lines[0].split(' ')
|
|
nhash=int(header[0]); names=set(header[1:]) # Makes deduping way easier
|
|
medals={}
|
|
if ':' in lines[-1]: # Since this isn't used in any other line, it's safe
|
|
medals={int(n.split(':')[0]):int(n.split(':')[1]) for n in lines.pop(-1).split(' ')}
|
|
stuff={k:parse_conf_line(v) for k,v in zip(tables.displays_str[:-1],lines[1:])}
|
|
conf[nhash]={'names':names}|stuff|{'medals':medals}
|
|
update_all_talents()
|
|
global active_profile
|
|
active_profile=nhash
|
|
|
|
def unparse_conf_line(line):
|
|
return ' '.join([str(n[0])+'/'+str(n[1]) for n in line])
|
|
|
|
def save_conf():
|
|
write=''
|
|
for k,v in conf.items():
|
|
write+=' '.join([str(k)]+list(v['names']))+'\n'
|
|
for n in map(lambda x:unparse_conf_line(v[x]),tables.displays_str[:-1]):
|
|
write+=n+'\n'
|
|
write+=' '.join(f'{h}:{m}' for h,m in v['medals'].items())+'\n'
|
|
write+='\n'
|
|
write=write[:-2]
|
|
open(confp,'w').write(write)
|
|
|
|
def update_all_talents():
|
|
for profile in conf:
|
|
cl=dict(enumerate(conf[profile]['classes']))
|
|
t=defaults['talents'].copy()
|
|
conf[profile]['talents']=t
|
|
for n in range(1,7):
|
|
t[n]=update_talent(n,cl)
|
|
|
|
def update_talents(clas,profile=None):
|
|
if profile is None: profile=active_profile
|
|
cl=dict(enumerate(conf[profile]['classes']))
|
|
for talent in tables.clas[clas][1:]: # Don't care about courage
|
|
conf[profile]['talents'][talent]=update_talent(talent,cl)
|
|
|
|
def update_talent(talent,classes):
|
|
a=enumerate(tables.clas); b=filter(lambda x:talent in x[1],a); c=list(map(lambda x:x[0],b)); d=filter(lambda x:x[0] in c,classes.items()); rel_classes=list(map(lambda x:x[1],d))
|
|
ranks,caps=[],[]
|
|
for n in rel_classes: ranks.append(n[0]); caps.append(n[1])
|
|
ranks,caps=sorted(ranks),sorted(caps)
|
|
lon=len(ranks)//2+len(ranks)%2
|
|
return (ranks[-lon],caps[-lon])
|
|
|
|
def read_code(code,name=None,save=None):
|
|
if name is None: name=active_profile; save2=True
|
|
else: save2=False # Don't save spoofed codes unless explicitly told
|
|
if save is None: save=save2
|
|
data=list(decode.decode(code,name))
|
|
check,validator=data.pop(-1),data.pop(-1)
|
|
talent=data.pop(5)
|
|
hero=data[0:6]
|
|
show=list(map(lambda x:x[0][x[1]],zip(tables.displays,hero)))
|
|
rank,cap=data[6:8]
|
|
cap=tables.cap[cap]
|
|
if rank==12: cap=12
|
|
data=data[8:]
|
|
if rank<9: show[-1]='Hidden' # Don't spoil the talent
|
|
print(show)
|
|
print(str(rank)+'/'+str(cap))
|
|
print(data) # Extract medals and save them
|
|
medals=data[1:-1]
|
|
if not check: print('No name given, code validation not performed')
|
|
else:
|
|
if check!=validator: print('code did not validate:',validator,check); return
|
|
else: # Do not save medals unless validation passed
|
|
if any(medals): conf[active_profile]['medals'][shared.muxhero(hero[0],hero[3],hero[4])]=shared.muxmedals(*medals)
|
|
print('code validated')
|
|
for thing in zip(tables.displays_str[:-1],hero):
|
|
saved=conf[active_profile][thing[0]][thing[1]]
|
|
saved=(max(saved[0],rank),max(saved[1],cap))
|
|
conf[active_profile][thing[0]][thing[1]]=saved
|
|
update_talents(hero[0])
|
|
if not save: return
|
|
save_conf()
|
|
|
|
def mass_add():
|
|
a=input('>')
|
|
while a:
|
|
try: read_code(a,false)
|
|
except: pass
|
|
a=input('>')
|
|
save_conf()
|
|
|
|
def generate(officer=None,name=None):
|
|
pro=conf[active_profile]
|
|
if name is None: name=active_profile
|
|
if officer is None: officer=input('> ')
|
|
pieces=officer.split('/')
|
|
c=pieces[0].lower()
|
|
if 'mav' in c:
|
|
gun=c[0]
|
|
pieces[0]='mav'
|
|
elif 'wm' in c:
|
|
gun=c[0:2]
|
|
pieces[0]='wm'
|
|
else: gun=''
|
|
out=[]
|
|
if len(pieces)==6: rank=cap=int(pieces.pop(-1))
|
|
else: rank,cap=12,12
|
|
try:
|
|
for n in zip(pieces,tables.displays_nogun,tables.displays_nogun_str):
|
|
ind=lookup_piece(n[0],n[1],n[2])
|
|
rank,cap=min(rank,pro[n[2]][ind][0]),min(cap,pro[n[2]][ind][1])
|
|
if 0 in [rank,cap]: print(f"{n[0]} is locked, go unlock it first.")
|
|
out.append(ind)
|
|
if gun: out.insert(1,lookup_piece(gun,tables.guns,'guns',True))
|
|
else: out.insert(1,tables.gunmaps_wrapped[out[0]])
|
|
except LookupException as ex: print(ex); return
|
|
rank,cap=min(rank,pro['guns'][out[1]][0]),min(cap,pro['guns'][out[1]][1]) # Derp2, forgot to actually cap on gun
|
|
if 0 in [rank,cap]: return # Already warned about locks earlier.
|
|
cap=(cap-(cap<10)+(cap>10))//3 # Derp
|
|
try: out[-1]=tables.clas[out[0]].index(out[-1])
|
|
except ValueError: print(f"{tables.classes[out[0]]} can't get talent {tables.talent[out[-1]]}, options are {list(map(lambda x:tables.talent[x],tables.clas[out[0]]))}"); return
|
|
muxed=shared.muxhero(out[0],out[3],out[4])
|
|
meds=[]
|
|
if muxed in pro['medals']: meds=shared.muxmedals(pro['medals'][muxed])
|
|
return encode.encode(name,*out,rank,cap,0,*meds)
|
|
|
|
def lookup_piece(piece,table,name,partial=False):
|
|
if len(piece)<3: p=piece.upper()
|
|
else: p=piece.capitalize() # Bloody Americans
|
|
if not partial:
|
|
try: return table.index(p)
|
|
except ValueError: raise LookupException(f"Couldn't resolve {piece} from {name}")
|
|
res=list(filter(lambda x:p in x,table))
|
|
if len(res)!=1: raise LookupException(f"Couldn't resolve {piece} with partial match from {name} (found {len(res)} possibilities)") # Ambiguous lookups not allowed
|
|
return table.index(res[0])
|
|
class LookupException(Exception): pass
|
|
|
|
def ranks():
|
|
for n,m in zip(tables.displays_str,tables.displays):
|
|
a=zip(m,conf[active_profile][n])
|
|
for o in a:
|
|
print(f'{o[0]}: {o[1][0]}/{o[1][1]}',end=', ')
|
|
print()
|
|
|
|
load_conf()
|
|
|
|
|
|
if __name__=="__main__": print("Please open an interpreter and use 'import codegen', this script doesn't serve as an entrypoint on its own yet.")
|