olivebot/utils/storer.py

32 lines
831 B
Python
Raw Normal View History

2022-06-20 11:05:10 +08:00
import codecs
import os
from threading import Thread
import time
from core.interact import Interact
2022-06-20 11:05:10 +08:00
FILE_URL = "datas/data-" + time.strftime("%Y%m%d%H%M%S") + ".csv"
def __write_to_file(text):
if not os.path.exists("datas"):
os.mkdir("datas")
file = codecs.open(FILE_URL, 'a', 'utf-8')
file.write(text + "\n")
file.close()
def storage_live_interact(interact: Interact):
interact_type = interact.interact_type
user = interact.data["user"].replace(',', '')
msg = interact.data["msg"].replace(',', '')
2022-06-20 11:05:10 +08:00
msg_type = {
0: '主播',
1: '发言',
2: '进入',
3: '送礼',
4: '关注'
}
timestamp = int(time.time() * 1000)
Thread(target=__write_to_file, args=["%s,%s,%s,%s\n" % (timestamp, msg_type[interact_type], user, msg)]).start()