From abd4237002b93d3d449a12f387dda18f2d0f19f5 Mon Sep 17 00:00:00 2001
From: Mike Smith <grimbough@gmail.com>
Date: Tue, 17 Oct 2017 13:54:58 +0200
Subject: [PATCH] updated code to print hostname
---
cluster_setup/hpc_example.py | 41 ++++++++++++++++++++++++++++++++++++
1 file changed, 41 insertions(+)
create mode 100755 cluster_setup/hpc_example.py
diff --git a/cluster_setup/hpc_example.py b/cluster_setup/hpc_example.py
new file mode 100755
index 0000000..d108110
--- /dev/null
+++ b/cluster_setup/hpc_example.py
@@ -0,0 +1,41 @@
+#!/usr/bin/python
+
+import sys, getopt, resource, time, platform
+
+def main(argv):
+ wait_time = 10
+ list_length = 0
+
+ ## handle the arguments
+ try:
+ opts, args = getopt.getopt(argv,"ht:l:",["time=","length="])
+ except getopt.GetoptError as err:
+ print str(err)
+ print 'python_example.py -t <wait time> -l <list length>'
+ sys.exit(2)
+ for opt, arg in opts:
+ if opt == '-h':
+ print 'python_example.py -t <wait time> -l <list length>'
+ sys.exit()
+ elif opt in ("-t", "--time"):
+ wait_time = max(10, float(arg) );
+ elif opt in ("-l", "--length"):
+ list_length = max(0, int(arg) )
+
+ print 'Current host is:', platform.node()
+ print 'Wait time is:', wait_time, 'seconds'
+ print 'List length is:', list_length
+
+ ## create a list of the required length
+ list = []
+ for i in range(0, list_length ):
+ list.append(1)
+
+ ## wait for the specified number of seconds
+ time.sleep( wait_time )
+
+ ## print the maximum memory usage to the screen
+ print 'Memory usage:', resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1000.0, 'MB'
+
+if __name__ == "__main__":
+ main(sys.argv[1:])
--
GitLab