Changeset 2427 for TOOLS/ConsoGENCMIP6/bin/plot_store.py

Timestamp:

03/02/15 17:43:15 (9 years ago)

Author:

labetoulle

Message:

Move common functions to common module

File:

• TOOLS/ConsoGENCMIP6/bin/plot_store.py (modified) (7 diffs)

TOOLS/ConsoGENCMIP6/bin/plot_store.py

@@ -40 +40 @@
   #---------------------------------------
   def fill_data(self, filein):
-    data = np.genfromtxt(
-      filein,
-      skip_header=1,
-      converters={0: string_to_date,
-                  1: str,
-                  2: string_to_size_unit,
-                  3: str},
-      missing_values="nan",
-    )
+    """
+    """
+    try:
+      data = np.genfromtxt(
+        filein,
+        skip_header=1,
+        converters={0: string_to_date,
+                    1: str,
+                    2: string_to_size_unit,
+                    3: str},
+        missing_values="nan",
+      )
+    except:
+      print("Empty file {}".format(filein))
+      exit(1)
 
     for date, login, dirsize, dirname in data:
@@ -161 +167 @@
   plt.figtext(x=0.95, y=0.93, s=tot_label, backgroundcolor="linen",
               ha="right", va="bottom", fontsize="small")
-
-
-########################################
-def plot_save(img_name):
-  """
-  """
-  dpi = 200.
-
-  with PdfPages(img_name) as pdf:
-    pdf.savefig(dpi=dpi)
-
-    # pdf file's metadata
-    d = pdf.infodict()
-    d["Title"]   = "Occupation GENCMIP6 sur STORE par login"
-    d["Author"]  = "plot_bilan.py"
-    # d["Subject"] = "Time spent over specific commands during create_ts \
-    #                 jobs at IDRIS and four configurations at TGCC"
-    # d["Keywords"] = "bench create_ts TGCC IDRIS ncrcat"
-    # d["CreationDate"] = dt.datetime(2009, 11, 13)
-    # d["ModDate"] = dt.datetime.today()
 
 
@@ -220 +206 @@
 
   import matplotlib.pyplot as plt
-  from matplotlib.backends.backend_pdf import PdfPages
+  # from matplotlib.backends.backend_pdf import PdfPages
 
   if not args.show:
@@ -227 +213 @@
   # ... Files and directories ...
   # -----------------------------
-  file_param = get_last_file(DIR["SAVE"], OUT["PARAM"])
-  file_utheo = get_last_file(DIR["SAVE"], OUT["UTHEO"])
-  file_store = get_last_file(DIR["SAVE"], OUT["STORE"])
-  img_name = "bilan.pdf"
+  (file_param, file_utheo, file_data) = \
+      get_input_files(DIR["SAVEDATA"],
+                      [OUT["PARAM"], OUT["UTHEO"], OUT["STORE"]])
+
+  img_name = "store"
+  today = os.path.basename(file_param).strip(OUT["PARAM"])
 
   if args.verbose:
     print(file_param)
     print(file_utheo)
-    print(file_store)
+    print(file_data)
     print(img_name)
+    print(today)
 
   # .. Get project info ..
@@ -247 +236 @@
   # =======================
   stores = StoreDict()
-  stores.fill_data(file_store)
+  stores.fill_data(file_data)
 
   # .. Extract data depending on C.L. arguments ..
@@ -301 +290 @@
   # ... Save figure ...
   # -------------------
-  plot_save(os.path.join(DIR["PLOT"], img_name))
+  plot_save(img_name, today, "Occupation GENCMIP6 sur STORE par login")
 
   # ... Publish figure on dods ...
@@ -311 +300 @@
     plt.show()
 
-  exit()
+  exit(0)