tfscript
/
tfscript
2
0
Fork 0
Code Issues 5 Pull Requests Projects 1 Releases Wiki Activity

sort of works with OOP

class_based_refactor
Nicholas Hope 2022-08-21 21:21:09 -04:00
parent 9ba82add22
commit c3b816c532
1 changed files with 38 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
src/tfscript/cli.py
View File

@ -1,13 +1,13 @@
"""
'''
Command line module for making Team Fortress 2 macro scripts from
YAML source code.
"""
'''
__all__ = ['parseFile']
__author__ = "Nicholas Hope <tfscript@nickhope.world"
__date__ = "26 August 2022"
__version__ = "1.0"
__copyright__ = "Copyright © 2022 Nicholas Hope. See LICENSE for details."
__author__ = 'Nicholas Hope <tfscript@nickhope.world'
__date__ = '26 August 2022'
__version__ = '1.0'
__copyright__ = 'Copyright © 2022 Nicholas Hope. See LICENSE for details.'
# Standard libraries
from sys import stderr
@ -31,38 +31,38 @@ from tfscript import verify
from tfscript import writing
args = {}
targetDir = ""
targetDir = ''
def parseFile(inputFile) -> (dict, dict):
"""Parse, verify, and do the conversion."""
'''Parse, verify, and do the conversion.'''
config = yaml.safe_load(inputFile)
# See verify.py
config, aliases = verify.verifyConfig(config)
if "errors" in config:
for cclass, messages in config["errors"].items():
print(f"Error in {cclass}:")
config, defaults = verify.verifyConfig(config)
if 'errors' in config:
for cclass, messages in config['errors'].items():
print(f'Error in {cclass}:')
for msg in messages:
print(f" {msg}")
print(f' {msg}')
return None, None
else:
return config, aliases
return config, defaults
def parseConfig(config, defaults):
"""With validated data structure, write out all the files."""
'''With validated data structure, write out all the files.'''
global args
global targetDir
if isdir(targetDir) == False:
mkdir(targetDir)
if args.debug:
print( f"DEBUG: Created directory {targetDir}", file=stderr)
print( f'DEBUG: Created directory {targetDir}', file=stderr)
tempsAndReals = {}
if defaults is not None:
stringToWrite = tfscript.makeCFG(defaults, default=True)
replaceDict = writing.writeOutput(stringToWrite, "default", args)
replaceDict = writing.writeOutput(stringToWrite, 'default', args)
tempsAndReals.update(replaceDict)
for currentClass in config:
@ -76,56 +76,56 @@ def parseConfig(config, defaults):
def parseCLI():
# Handle command line
parser = argparse.ArgumentParser(
description="Parse YAML file and produce TF2 config script."
description='Parse YAML file and produce TF2 config script.'
)
parser.add_argument( '-d', '--debug', action='store_true',
help="Enable debugging messages.")
help='Enable debugging messages.')
parser.add_argument( '-n', '--dry-run', action='store_true',
help="Parse input file, but don't write anything.")
help='Parse input file, but don\'t write anything.')
parser.add_argument( '-f', '--force', action='store_true',
help="Force tfscript to continue until catastrophic failure")
help='Force tfscript to continue until catastrophic failure')
parser.add_argument( '-D', '--directory', action='store', type=str,
help="Change output directory")
help='Change output directory')
# positional argument: first non-hyphenated argument is input file
parser.add_argument( 'infile', type=argparse.FileType('r'),
help='File containing YAML to convert.')
return parser
def getTargetDir(systemName):
if systemName == "Darwin":
if systemName == 'Darwin':
if float( '.'.join( GetOSRelease().split('.')[0:2] ) ) >= 10.15:
warn(
"As of macOS Catalina (v10.15), 32-bit applications "
"like TF2 do not run. tfscript will run, but you can't run TF2 "
"on this system",
'As of macOS Catalina (v10.15), 32-bit applications '
'like TF2 do not run. tfscript will run, but you can\'t run TF2 '
'on this system',
category=RuntimeWarning )
return expanduser("~/Library/Application Support/Steam")
return expanduser('~/Library/Application Support/Steam')
elif systemName == "Windows":
elif systemName == 'Windows':
# oh god why do we have to use the registry
accessReg = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
accessKey = OpenKey(accessReg, "SOFTWARE\\WOW6432Node\\Valve\\Steam")
accessKey = OpenKey(accessReg, 'SOFTWARE\\WOW6432Node\\Valve\\Steam')
keyNum = 0
while True:
try:
accessSubkeyName, data, _ = EnumValue(accessKey, keyNum)
if accessSubkeyName == "InstallPath":
if accessSubkeyName == 'InstallPath':
return data
except EnvironmentError:
break
keyNum += 1
return None
elif systemName == "Linux":
return expanduser("~/.local/Steam")
elif systemName == 'Linux':
return expanduser('~/.local/Steam')
elif systemName == "Java":
warn("Java-based OSes are not supported yet by tfscript.", category=RuntimeWarning)
elif systemName == 'Java':
warn('Java-based OSes are not supported yet by tfscript.', category=RuntimeWarning)
return None
def main() -> int:
""" Command line interface. """
''' Command line interface. '''
global args
global targetDir
parser = parseCLI()
@ -139,11 +139,11 @@ def main() -> int:
targetDir = getTargetDir(systemName)
if targetDir is not None:
# Supported OS: add steamapps path
targetDir += normpath("/steamapps/common/Team Fortress 2/tf/cfg") + dirsep
targetDir += normpath('/steamapps/common/Team Fortress 2/tf/cfg') + dirsep
elif args.force:
# Unsupported OS but -f specified
if args.debug:
print("DEBUG: forced to continue, output set to current directory", file=stderr)
print('DEBUG: forced to continue, output set to current directory', file=stderr)
targetDir = '.'
else:
# Unsupported OS and not forced to continue
@ -160,5 +160,5 @@ def main() -> int:
return 0
if __name__ == "__main__":
if __name__ == '__main__':
exit(main())