From 4c7c9a310ff5c7185b8ae095d39e0bfdba4a1d02 Mon Sep 17 00:00:00 2001
From: pengtao <pengtao@kingsome.cn>
Date: Mon, 4 Nov 2019 11:30:49 +0800
Subject: [PATCH] add tags field

---
 taptap/tap_split_tags.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/taptap/tap_split_tags.py b/taptap/tap_split_tags.py
index 8a9abcb..ee9f60b 100644
--- a/taptap/tap_split_tags.py
+++ b/taptap/tap_split_tags.py
@@ -17,7 +17,7 @@ class SplitTapTags:
         self.day = day
 
 
-    def run(self):
+    def get_data(self):
         all = dict()
         for types in self.type:
             all[types] = {}
@@ -38,6 +38,26 @@ class SplitTapTags:
                         except Exception:
                             log.error(f"split {line} failed!", exc_info=True)
         print(all)
+        return all
+
+    def write2db(self, data):
+        db_data = {}
+        tables_name = "tap_tags_counts"
+        for key in data.keys():
+            for k in data[key]:
+                db_data['type'] = key
+                db_data['name'] = k
+                db_data['date'] = self.day
+                db_data['nums'] = data[key][k]
+                try:
+                    self.db_conn.insert(tables_name, db_data)
+                except Exception:
+                    log.error(f"insert {db_data} failed!", exc_info=True)
+                    
+
+    def run(self):
+        data = self.get_data()
+        self.write2db(data)
 
 
 def main():