124 lines
3.4 KiB
Python
124 lines
3.4 KiB
Python
# coding: utf8
|
|
|
|
# toutes les chaines sont en unicode (même les docstrings)
|
|
from __future__ import unicode_literals
|
|
|
|
# from pprint import pprint
|
|
from rocketchat_API.rocketchat import RocketChat
|
|
import json
|
|
import os
|
|
import random
|
|
from datetime import datetime, timedelta
|
|
from common.rocketchathelper import getAllChannels, getAllMessages, Connection
|
|
from common.savehelper import save
|
|
from common.charthelper import createElement, getColor, setTsunamyInfo, getTsunamy, Tsunami
|
|
|
|
begin = datetime.now()
|
|
end = datetime.now()
|
|
|
|
def main():
|
|
rocket = Connection()
|
|
nbElements = 30
|
|
|
|
global begin, end
|
|
|
|
labels = [None] * nbElements
|
|
messagesByChannel = []
|
|
usersByChannel = []
|
|
messagesDataTsunamy = {
|
|
Tsunami.GLOBAL: [0] * nbElements,
|
|
Tsunami.PROJECT: [0] * nbElements,
|
|
Tsunami.DEMOCRACY: [0] * nbElements,
|
|
Tsunami.ECOLOGY: [0] * nbElements,
|
|
Tsunami.TECHNOLOGY: [0] * nbElements,
|
|
}
|
|
usersGlobal = []
|
|
|
|
now = datetime.now()
|
|
date = datetime(now.year, now.month, now.day, 0,0,0)
|
|
|
|
info = {
|
|
"updated": "updated {:02}/{:02}/{:04}".format(now.day, now.month, now.year),
|
|
"labels": labels,
|
|
"messagesByChannel": messagesByChannel,
|
|
"usersByChannel": usersByChannel,
|
|
"messagesByTsunamy": [
|
|
createElement("global", getColor(), messagesDataTsunamy[Tsunami.GLOBAL]),
|
|
createElement("project", getColor(), messagesDataTsunamy[Tsunami.PROJECT]),
|
|
createElement("democratie", getColor(), messagesDataTsunamy[Tsunami.DEMOCRACY]),
|
|
createElement("ecologie", getColor(), messagesDataTsunamy[Tsunami.ECOLOGY]),
|
|
createElement("technologie", getColor(), messagesDataTsunamy[Tsunami.TECHNOLOGY])
|
|
],
|
|
"usersGlobal": usersGlobal
|
|
}
|
|
|
|
uniqueUserGlobal = [None] * nbElements
|
|
|
|
for channel in getAllChannels(rocket):
|
|
dataMess = []
|
|
dataUsers = []
|
|
print( channel['name'] )
|
|
begin = date - timedelta(nbElements)
|
|
end = begin + timedelta(nbElements)
|
|
|
|
tsunamy = getTsunamy(channel)
|
|
|
|
messages = getAllMessages(rocket, channel['_id'], begindate= begin, enddate= end)
|
|
end = begin + timedelta(1)
|
|
|
|
for id in range(0, nbElements):
|
|
if uniqueUserGlobal[id] == None:
|
|
uniqueUserGlobal[id] = []
|
|
labels[id] = begin.strftime("%x")
|
|
print(f"\t{labels[id]}")
|
|
|
|
resultMess = list(filter(filter_message, messages))
|
|
length = len(resultMess)
|
|
dataMess.append(length)
|
|
|
|
if length > 0:
|
|
setTsunamyInfo(tsunamy, messagesDataTsunamy, id, length)
|
|
|
|
users = []
|
|
for mess in resultMess:
|
|
users.append(mess['u']['_id'])
|
|
uniqueUserGlobal[id].append(mess['u']['_id'])
|
|
usersDistinct = set(users)
|
|
dataUsers.append(len(usersDistinct))
|
|
else:
|
|
dataUsers.append(0)
|
|
|
|
begin = end
|
|
end = begin + timedelta(1)
|
|
|
|
color = getColor()
|
|
messageByChannel = createElement(channel['name'], color,dataMess)
|
|
userByChannel = createElement( channel['name'], color,dataUsers)
|
|
|
|
messagesByChannel.append(messageByChannel)
|
|
usersByChannel.append(userByChannel)
|
|
|
|
for id in range(0, nbElements):
|
|
uniqueUserGlobal[id] = len(set(uniqueUserGlobal[id]))
|
|
|
|
userGlobal = createElement( 'global', 'red', uniqueUserGlobal)
|
|
usersGlobal.append(userGlobal)
|
|
|
|
save(info, "chat_stat_daily")
|
|
|
|
def filter_message(mess):
|
|
if 't' in mess:
|
|
return False
|
|
|
|
date = datetime.strptime(mess["ts"], "%Y-%m-%dT%H:%M:%S.%fZ")
|
|
|
|
if date < begin:
|
|
return False
|
|
|
|
if end < date:
|
|
return False
|
|
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
main() |