32 lines
831 B
Python
32 lines
831 B
Python
import codecs
|
||
import os
|
||
from threading import Thread
|
||
import time
|
||
|
||
from core.interact import Interact
|
||
|
||
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(',', '，')
|
||
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()
|