save plt to file

analysis_db.py

@@ -38,28 +38,32 @@ def read_db(db_name, tb_name='tb_section', section_id=''):
         debug_print('section_id = ', section_id, ', _sample_id = ', _sample_id, ', _sample_end_id = ',
                     _sample_end_id)
 
+        date_list = []
         max_list = []
         min_list = []
         acc_list = []
-        sample_sql = "SELECT _id, max, min, acc_max_dx, acc_max_dy, acc_max_dz FROM sample_table WHERE _id >= " + str(_sample_id) + ' AND ' + ' _id <= ' + str(_sample_end_id)
+        sample_sql = "SELECT _id, date, max, min, acc_max_dx, acc_max_dy, acc_max_dz FROM sample_table WHERE _id >= " + str(_sample_id) + ' AND ' + ' _id <= ' + str(_sample_end_id)
         sample_cursor = conn.execute(sample_sql)
         for sample_row in sample_cursor:
             _id = sample_row[0]
-            max = sample_row[1]
-            min = sample_row[2]
-            acc_max = sample_row[3] + sample_row[4] + sample_row[5]
-            debug_print('_id = ', _id,  'sample_row[3] = ', str(sample_row[3]))
+            date = sample_row[1]
+            max = sample_row[2]
+            min = sample_row[3]
+            acc_max = sample_row[4] + sample_row[5] + sample_row[6]
+            debug_print('_id = ', _id,  'date = ', format_date(date))
+
+            date_list.append(format_date(date))
             max_list.append(max)
             min_list.append(min)
             acc_list.append(acc_max)
 
         # debug_print('max_list = ', str(max_list))
         # debug_print('min_list = ', str(min_list))
-        debug_print('acc_list = ', str(acc_list))
+        # debug_print('acc_list = ', str(acc_list))
         # draw(max_list)
         title_str = str(section_id) + ', ' + str(format_date(section_date)) + ', ' + str(format_date(section_end_date))
         # draw_two(title_str, max_list, min_list)
-        draw_three(title_str, max_list, min_list, acc_list)
+        draw_three(title_str, date_list, max_list, min_list, acc_list)
 
     conn.close()
 

debug.py

@@ -1,5 +1,5 @@
-DEBUG = True
-# DEBUG = False
+# DEBUG = True
+DEBUG = False
 
 
 def debug_print(self, *args, sep=' ', end='\n', file=None):

draw_db.py

@@ -1,4 +1,5 @@
 import matplotlib.pyplot as plt
+from debug import debug_print
 
 
 def draw(x_points=None, y_points=None):
@@ -19,13 +20,24 @@ def draw_two(title='', x_list1=None, x_list2=None):
     plt.show()
 
 
-def draw_three(title='', x_list1=None, x_list2=None, x_list3=None):
+def draw_three(title='', date_list=None, x_list1=None, x_list2=None, x_list3=None):
+    debug_print('draw_three, date_list = ', str(date_list))
+
     fig, ax = plt.subplots()
-    ax.plot(x_list1)
-    ax.plot(x_list2)
-    ax.plot(x_list3)
+    if date_list is None:
+        ax.plot(x_list1)
+        ax.plot(x_list2)
+        ax.plot(x_list3)
+    else:
+        ax.plot(date_list, x_list1)
+        ax.plot(date_list, x_list2)
+        ax.plot(date_list, x_list3)
     ax.legend()
+    plt.gcf().autofmt_xdate()
 
     plt.ylim(ymin=0, ymax=120)
     plt.title(title)
-    plt.show()
+    # plt.show()
+
+    file_uri = '.doc/' + title + '.png'
+    plt.savefig(file_uri)