From 12ff99eb3213569a776363f630c3767efd596da6 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Thu, 6 Dec 2018 18:27:43 -0700
Subject: [PATCH] Set LogDir for non-quick tests

---
 .bin/Scripts/functions/hw_diags.py | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/.bin/Scripts/functions/hw_diags.py b/.bin/Scripts/functions/hw_diags.py
index a76666ab..e1b97df9 100644
--- a/.bin/Scripts/functions/hw_diags.py
+++ b/.bin/Scripts/functions/hw_diags.py
@@ -180,8 +180,7 @@ class State():
     self.devs = []
     self.finished = False
     self.panes = {}
-    # TODO Switch to LogDir
-    self.progress_out = '{}/progress.out'.format(global_vars['TmpDir'])
+    self.progress_out = '{}/progress.out'.format(global_vars['LogDir'])
     self.quick_mode = False
     self.started = False
     self.tests = {
@@ -221,6 +220,16 @@ class State():
     for k in ['Result', 'Started', 'Status']:
       self.tests['Prime95 & Temps'][k] = False if k == 'Started' else ''
 
+    # Update LogDir
+    if not self.quick_mode:
+      global_vars['LogDir'] = '{}/Logs/{}_{}'.format(
+        global_vars['Env']['HOME'],
+        get_ticket_number(),
+        time.strftime('%Y-%m-%d_%H%M_%z'))
+      os.makedirs(global_vars['LogDir'], exist_ok=True)
+      global_vars['LogFile'] = '{}/Hardware Diagnostics.log'.format(
+        global_vars['LogDir'])
+
     # Add block devices
     cmd = ['lsblk', '--json', '--nodeps', '--paths']
     result = run_program(cmd, check=False)
@@ -703,6 +712,15 @@ def run_mprime_test(state):
     function=save_average_temp, cs='Done',
     sensor_data=_sensor_data, temp_label='Cooldown')
 
+  # Move logs to Ticket folder
+  for item in os.scandir(global_vars['TmpDir']):
+    try:
+      shutil.move(item.path, global_vars['LogDir'])
+    except Exception:
+      print_error('ERROR: Failed to move "{}" to "{}"'.format(
+        item.path,
+        global_vars['LogDir']))
+
   # Check results
   # TODO