napcok/calamares
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PEPify testmodule.py

Browse Source
This commit is contained in:
Teo Mrnjavac 2014-07-29 13:52:55 +02:00
parent b988c3ad72
commit 637242776c
1 changed files with 29 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/modules/testmodule.py
View File

@ -25,61 +25,62 @@ import yaml
try:
import libcalamares
except ImportError:
print( "Failed to import libcalamares. Make sure then PYTHONPATH environment variable includes the dir where libcalamares.so is installed." )
print("Failed to import libcalamares. Make sure then PYTHONPATH "
"environment variable includes the dir where libcalamares.so is "
"installed.")
print()
raise
class Job:
def __init__( self, workingPath, doc ):
self.module_name = doc[ "name" ]
self.pretty_name = "Testing job " + doc[ "name" ]
self.working_path = workingPath
self.configuration = doc[ "configuration" ]
def setprogress( self, progress ):
print( "Job set progress to {}%.".format( progress * 100 ) )
def __init__(self, working_path, doc):
self.module_name = doc["name"]
self.pretty_name = "Testing job " + doc["name"]
self.working_path = working_path
self.configuration = doc["configuration"]
def setprogress(self, progress):
print("Job set progress to {}%.".format(progress * 100))
def main():
parser = argparse.ArgumentParser()
parser.add_argument( "moduledir",
help = "Dir containing the Python module" )
parser.add_argument( "globalstorage_yaml", nargs = "?",
help = "A yaml file to initialize GlobalStorage" )
parser.add_argument("moduledir",
help="Dir containing the Python module")
parser.add_argument("globalstorage_yaml", nargs="?",
help="A yaml file to initialize GlobalStorage")
args = parser.parse_args()
print( "Testing module in: " + args.moduledir )
print("Testing module in: " + args.moduledir)
confpath = os.path.join( args.moduledir, "module.conf" )
with open( confpath ) as f:
doc = yaml.load( f )
confpath = os.path.join(args.moduledir, "module.conf")
with open(confpath) as f:
doc = yaml.load(f)
if doc[ "type" ] != "job" or doc[ "interface" ] != "python":
print( "Only Python jobs can be tested." )
if doc["type"] != "job" or doc["interface"] != "python":
print("Only Python jobs can be tested.")
return 1
libcalamares.job = Job( args.moduledir, doc )
libcalamares.job = Job(args.moduledir, doc)
libcalamares.globalstorage = libcalamares.GlobalStorage()
# if a file for simulating globalStorage contents is provided, load it
if args.globalstorage_yaml:
with open( args.globalstorage_yaml ) as f:
doc = yaml.load( f )
with open(args.globalstorage_yaml) as f:
doc = yaml.load(f)
for key, value in doc.items():
libcalamares.globalStorage.insert( key, value )
libcalamares.globalStorage.insert(key, value)
scriptpath = os.path.abspath( args.moduledir )
sys.path.append( scriptpath )
scriptpath = os.path.abspath(args.moduledir)
sys.path.append(scriptpath)
import main
print( "Output from module:" )
print( main.run() )
print("Output from module:")
print(main.run())
return 0
if __name__ == "__main__":
sys.exit( main() )
sys.exit(main())