olivebot/utils/storer.py

30 lines
755 B
Python
Raw Normal View History

2022-06-20 11:05:10 +08:00
import codecs
import os
from threading import Thread
import time
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_type = interact[0]
user = interact[1].replace(',', '')
msg = interact[2].replace(',', '')
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()