From a740413f4e30342e1e5b46142b9ce9026089e1fd Mon Sep 17 00:00:00 2001
From: tursom <tursom@foxmail.com>
Date: Sat, 16 Mar 2024 15:00:45 +0800
Subject: [PATCH] make redis to config

---
 .gitignore |  3 +++
 sample.py  | 12 ++++++++++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index d3d675d..5d9e755 100644
--- a/.gitignore
+++ b/.gitignore
@@ -103,3 +103,6 @@ ENV/
 
 cookie.txt
 logs
+
+config.json
+
diff --git a/sample.py b/sample.py
index 69530a8..d0b7169 100644
--- a/sample.py
+++ b/sample.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 import asyncio
 import http.cookies
 import json
@@ -15,6 +14,11 @@ import redis
 import blivedm
 from blivedm.clients import ws_base
 
+with open("config.json") as conf_f:
+    conf = json.load(conf_f)
+
+    redis_conf = conf.get("redis", {})
+
 os.makedirs("logs", exist_ok=True)
 room_status_log = open("logs/room_status.jsonl", "a", encoding="utf-8")
 
@@ -24,7 +28,11 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger()
 
 room_list_key = "bilibili.live.danmu.room_list"
-r = redis.Redis(host="redis", port=6379, db=0)
+r = redis.Redis(
+    host=redis_conf.get("host", "127.0.0.1"),
+    port=redis_conf.get("port", 6379),
+    db=redis_conf.get("port", 0),
+)
 
 TEST_ROOM_IDS = [int(room_id) for room_id in set(r.lrange(room_list_key, 0, -1))]
 clients = {}