From 97fe502889175d9c63c2f323e5e5933e8caa9108 Mon Sep 17 00:00:00 2001
From: jannchie <jannchie@gmail.com>
Date: Thu, 24 Oct 2019 11:47:58 +0800
Subject: [PATCH] fix: filter

---
 biliob_analyzer/author_fans_watcher.py |  2 +-
 biliob_analyzer/db.py                  | 22 ++++++++++++++++++++++
 biliob_requests/db.py                  | 13 +++++++++++++
 3 files changed, 36 insertions(+), 1 deletion(-)
 create mode 100644 biliob_analyzer/db.py
 create mode 100644 biliob_requests/db.py

diff --git a/biliob_analyzer/author_fans_watcher.py b/biliob_analyzer/author_fans_watcher.py
index 5228ba1..9cdc290 100644
--- a/biliob_analyzer/author_fans_watcher.py
+++ b/biliob_analyzer/author_fans_watcher.py
@@ -36,7 +36,7 @@ def __insert_event(self, delta_rate, d_daily, author, info, date):
         cause = {'type': 'video'}
         for each_v in videos:
             # 相差一日之内
-            if type(each_v['datetime']) == 'str':
+            if type(each_v['datetime']) == str:
                 pass
             elif (date - each_v['datetime']).days >= -1 and (date - each_v['datetime']).days <= 7:
                 temp_video['aid'] = each_v['aid']
diff --git a/biliob_analyzer/db.py b/biliob_analyzer/db.py
new file mode 100644
index 0000000..19d4ca1
--- /dev/null
+++ b/biliob_analyzer/db.py
@@ -0,0 +1,22 @@
+from pymongo import MongoClient
+import pymysql
+import redis
+import os
+import datetime
+env_dist = os.environ
+try:
+    client = MongoClient(env_dist.get("BILIOB_MONGO_URL"))
+    db = client['biliob']  # 获得数据库的句柄
+except Exception:
+    print('{} MongoDB链接失败'.format(datetime.datetime.now()))
+    os.system('systemctl restart mongod')
+
+redis_connect_string = env_dist.get("BILIOB_REDIS_CONNECTION_STRING")
+redis_list = redis_connect_string.split(":")
+pass_and_ip = redis_list[2].split('@')
+password = pass_and_ip[0]
+redis_ip = pass_and_ip[1]
+redis_port = redis_list[-1]
+pool = redis.ConnectionPool(password)  # 实现一个连接池
+redis_connection = redis.Redis(
+    host=redis_ip, port=redis_port, password=password)
diff --git a/biliob_requests/db.py b/biliob_requests/db.py
new file mode 100644
index 0000000..340a3f5
--- /dev/null
+++ b/biliob_requests/db.py
@@ -0,0 +1,13 @@
+from pymongo import MongoClient
+
+settings = {
+    'MINGO_USER': 'jannchie',
+    'MONGO_PSW': '141421',
+    'MINGO_HOST': '127.0.0.1'
+}
+# 链接mongoDB
+client = MongoClient(settings['MINGO_HOST'], 27017)
+# 数据库登录需要帐号密码
+client.admin.authenticate(settings['MINGO_USER'], settings['MONGO_PSW'])
+db = client['biliob']  # 获得数据库的句柄
+