diff --git a/1_single_core_job/single_core_task.py b/1_single_core_job/single_core_task.py index eb3087e3265c4142262dabd5921671ed8bfb9007..9e8e26d1f9b1a1ce66ac181168e6b900bd7d2f75 100644 --- a/1_single_core_job/single_core_task.py +++ b/1_single_core_job/single_core_task.py @@ -1,6 +1,10 @@ from datetime import datetime -import time + +import logging import os +import time + +logger = logging.getLogger(__name__) def main(): # Read environment variables. @@ -13,12 +17,12 @@ def main(): # This represents the calculations current_time = datetime.now() sleep_time = 60 - current_time.second - print('{} - Sleeping for {} seconds.'.format(current_time.strftime('%Y-%m-%d %H:%M:%S'), sleep_time)) + logger.info('%s - Sleeping for %d seconds.',current_time.strftime('%Y-%m-%d %H:%M:%S'), sleep_time) time.sleep(sleep_time) # Printing some things to standard output. - print('\nJob ID:\t\t\t{name}\nJob name:\t\t{id}\nAllocated cores:\t{cores}\nAllocated memory:\t{mem}'.format( - id=JOB_ID, name=JOB_NAME, cores=NUMBER_OF_CORES,mem=MAXIMUM_MEMORY)) + logger.info('\nJob ID:\t\t\t%s\nJob name:\t\t%s\nAllocated cores:\t%s\nAllocated memory:\t%s', + JOB_ID, JOB_NAME, NUMBER_OF_CORES,MAXIMUM_MEMORY) # Writing some output to a file based on the Slurm job id. output_file = '{}.txt'.format(JOB_ID) @@ -26,7 +30,8 @@ def main(): file.write('This file was created by the job {} with id {}\n'.format (JOB_NAME, JOB_ID)) - print('\nJob completed.') + logger.info('Job completed.') if __name__ == '__main__': + logging.basicConfig(level=logging.INFO) main()