renato97/twitch-highlight-detector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Seems finished

Browse Source
This commit is contained in:
Vitalii Lebedynskyi
2022-08-17 16:51:05 +03:00
parent 173fcc098f
commit fdfd2c6135
3 changed files with 121 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
clipper/analyser.py
View File

@@ -5,17 +5,18 @@ import matplotlib.pyplot as plt
from datetime import datetime
from clipper.chat import CHAT_DIVIDER
logger = logging.getLogger(__name__)
class ChatAnalyser:
def __init__(self, ignore_commands=True, ignored_users=["moobot", "nightbot"]):
def __init__(self, ignore_commands=True, ignored_users=None):
if ignored_users is None:
ignored_users = ["moobot", "nightbot"]
self.ignored_users = ignored_users
self.ignore_commands = ignore_commands
def run(self, chat_file, peaks_output_file, peaks_output_chart):
def run(self, chat_file, peaks_output_file, peaks_output_chart, start_time):
dates = self._read_message_dates(chat_file)
messages_per_minute = self._group_dates(dates)
peaks = self._find_peeks(messages_per_minute, peaks_output_file, peaks_output_chart)
@@ -31,19 +32,23 @@ class ChatAnalyser:
if not line:
break
message_data = line.split(CHAT_DIVIDER)
message_data = line.split("<~|~>")
if len(message_data) != 3:
# Wrong line format
continue
if message_data[1].lower() in self.ignore_commands:
if message_data[1].lower() in self.ignored_users:
continue
if self.ignore_commands and message_data[2].startswith("!"):
continue
date = message_data[0]
dates.append(self._parse_date(date))
try:
dates.append(self._parse_date(date))
except BaseException as e:
logger.error(e)
return dates
def _parse_date(self, date_str):
@@ -64,20 +69,39 @@ class ChatAnalyser:
def _find_peeks(self, messages_per_minute, peaks_output_file, peaks_output_chart):
y_coordinates = list(messages_per_minute.values())
x_coordinates = list(messages_per_minute.keys())
peak_indices = scipy.signal.find_peaks_cwt(np.array(y_coordinates), 1)
peak_indices = scipy.signal.find_peaks_cwt(np.array(y_coordinates), 0.5)
x_hours = [x.split(" ")[1] for x in x_coordinates]
fig, ax = plt.subplots()
ax.plot(x_hours, y_coordinates)
fig.autofmt_xdate()
plt.xlabel("Time")
plt.ylabel("Count")
ax.plot(range(0, len(y_coordinates), 1), y_coordinates)
plt.xlabel("Video Minutes")
plt.ylabel("Message count")
plt.title("Stream chat reaction")
plt.savefig(peaks_output_chart)
peak_values = [x_coordinates[index] for index in peak_indices]
start_time = None
if len(x_coordinates) > 0:
start_time = datetime.strptime(x_coordinates[0], "%Y-%m-%d %H:%M")
max_value = max(y_coordinates)
trash_hold_value = max_value * 0.75
filtered_values = [x_coordinates[index] for index in peak_indices if y_coordinates[index] > trash_hold_value]
with open(peaks_output_file, "w") as stream:
for peak in peak_values:
stream.writelines(f"{peak}\n")
for peak in filtered_values:
if start_time:
peak_time = datetime.strptime(peak, "%Y-%m-%d %H:%M")
diff = peak_time - start_time
minutes = divmod(diff.total_seconds() / 60, 60)
stream.writelines(f"{peak} -> {minutes}\n")
else:
stream.writelines(f"{peak}\n")
return peak_indices
if __name__ == "__main__":
anal = ChatAnalyser()
chat_file = "/Users/vetalll/Projects/Python/TwitchClipper/recorded/vovapain/17-08-2022_08-33-23/chat.txt"
out_file = "/Users/vetalll/Projects/Python/TwitchClipper/recorded/vovapain/17-08-2022_08-33-23/chat_peaks.txt"
out_hraph = "/Users/vetalll/Projects/Python/TwitchClipper/recorded/vovapain/17-08-2022_08-33-23/chat_chart.png"
anal.run(chat_file, out_file, out_hraph, datetime(2022, 8, 15, 20, 38, 49))