1st refacto
This commit is contained in:
Mickael
parent
20157b2cd3
commit
2c188c4ccf
1 changed files with 108 additions and 128 deletions
|
|
@ -20,145 +20,125 @@ def getColor():
|
|||
b = random.randrange(255)
|
||||
return 'rgb({:0},{:0},{:0})'.format(r,g,b)
|
||||
|
||||
rocket = RocketChat(cfg.rocket['user'], cfg.rocket['password'],
|
||||
server_url='https://coa.crapaud-fou.org')
|
||||
index = 0
|
||||
labels = [None] * 12
|
||||
messagesByChannel = []
|
||||
messagesByTsunamy = []
|
||||
usersByChannel = []
|
||||
messagesDataTsunamy = {
|
||||
"global": [0] * 12,
|
||||
"project": [0] * 12,
|
||||
"democraty": [0] * 12,
|
||||
"ecology": [0] * 12,
|
||||
"technology": [0] * 12,
|
||||
}
|
||||
usersGlobal = []
|
||||
def save(info):
|
||||
# Récupération du répertoire racine du repo
|
||||
rootFolder = os.path.join(os.path.dirname(__file__), '..')
|
||||
# Répertoire pour stocker le fichier de sortie
|
||||
dataFolder = os.path.join(rootFolder, 'public', 'data')
|
||||
|
||||
now = datetime.now()
|
||||
date = datetime(now.year, now.month, now.day, 0,0,0)
|
||||
statsFilePath = os.path.abspath(
|
||||
os.path.join(dataFolder, 'channelsstat.json'))
|
||||
with open(statsFilePath, "w") as file_write:
|
||||
json.dump(info, file_write)
|
||||
|
||||
info = {
|
||||
"updated": "updated {:02}/{:02}/{:04}".format(now.day, now.month, now.year),
|
||||
"labels": labels,
|
||||
"messagesByChannel": messagesByChannel,
|
||||
"usersByChannel": usersByChannel,
|
||||
"messagesByTsunamy": [{
|
||||
"label": "global",
|
||||
"backgroundColor": getColor(),
|
||||
"data": messagesDataTsunamy['global']
|
||||
},
|
||||
{
|
||||
"label": "projet",
|
||||
"backgroundColor": getColor(),
|
||||
"data": messagesDataTsunamy['project']
|
||||
},
|
||||
{
|
||||
"label": "democratie",
|
||||
"backgroundColor": getColor(),
|
||||
"data": messagesDataTsunamy['democraty']
|
||||
},
|
||||
{
|
||||
"label": "ecologie",
|
||||
"backgroundColor": getColor(),
|
||||
"data": messagesDataTsunamy['ecology']
|
||||
},
|
||||
{
|
||||
"label": "technologie",
|
||||
"backgroundColor": getColor(),
|
||||
"data": messagesDataTsunamy['technology']
|
||||
}],
|
||||
def createElement(label, color, data) :
|
||||
return {
|
||||
"label": label,
|
||||
"backgroundColor": color,
|
||||
"data": data
|
||||
}
|
||||
|
||||
def main():
|
||||
rocket = RocketChat(cfg.rocket['user'], cfg.rocket['password'],
|
||||
server_url='https://coa.crapaud-fou.org')
|
||||
index = 0
|
||||
labels = [None] * 12
|
||||
messagesByChannel = []
|
||||
usersByChannel = []
|
||||
messagesDataTsunamy = {
|
||||
"global": [0] * 12,
|
||||
"project": [0] * 12,
|
||||
"democraty": [0] * 12,
|
||||
"ecology": [0] * 12,
|
||||
"technology": [0] * 12,
|
||||
}
|
||||
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['global']),
|
||||
createElement("project", getColor(), messagesDataTsunamy['project']),
|
||||
createElement("democratie", getColor(), messagesDataTsunamy['democraty']),
|
||||
createElement("ecologie", getColor(), messagesDataTsunamy['ecology']),
|
||||
createElement("technologie", getColor(), messagesDataTsunamy['technology'])
|
||||
],
|
||||
"usersGlobal": usersGlobal
|
||||
}
|
||||
}
|
||||
|
||||
usersTest = [None] * 12
|
||||
usersTest = [None] * 12
|
||||
|
||||
while True:
|
||||
channels = rocket.channels_list(offset=index).json()
|
||||
totalChannels = channels['total']
|
||||
|
||||
for channel in channels['channels']:
|
||||
dataMess = []
|
||||
dataUsers = []
|
||||
pprint( channel['name'] )
|
||||
begin = date - monthdelta(12)
|
||||
end = begin + monthdelta(1)
|
||||
|
||||
tsunamy = getTsunamy(channel)
|
||||
|
||||
for id in range(0, 12):
|
||||
if usersTest[id] == None:
|
||||
usersTest[id] = []
|
||||
labels[id] = begin.strftime("%b %Y")
|
||||
begindate = begin.isoformat()
|
||||
enddate = end.isoformat()
|
||||
resultMess = rocket.channels_history(channel['_id'], oldest= begindate, latest=enddate, count= 10000).json()
|
||||
resultMess = list(filter(lambda mess: 't' not in mess, resultMess['messages']))
|
||||
lenght = len(resultMess)
|
||||
dataMess.append(lenght)
|
||||
|
||||
if lenght > 0:
|
||||
if tsunamy & Tsunami.GLOBAL:
|
||||
messagesDataTsunamy['global'][id] += lenght
|
||||
if tsunamy & Tsunami.PROJECT:
|
||||
messagesDataTsunamy['project'][id] += lenght
|
||||
if tsunamy & Tsunami.DEMOCRACY:
|
||||
messagesDataTsunamy['democraty'][id] += lenght
|
||||
if tsunamy & Tsunami.ECOLOGY:
|
||||
messagesDataTsunamy['ecology'][id] += lenght
|
||||
if tsunamy & Tsunami.TECHNOLOGY:
|
||||
messagesDataTsunamy['technology'][id] += lenght
|
||||
|
||||
users = []
|
||||
for mess in resultMess:
|
||||
users.append(mess['u']['_id'])
|
||||
usersTest[id].append(mess['u']['_id'])
|
||||
usersDistinct = set(users)
|
||||
dataUsers.append(len(usersDistinct))
|
||||
else:
|
||||
dataUsers.append(0)
|
||||
|
||||
begin = end
|
||||
while True:
|
||||
channels = rocket.channels_list(offset=index).json()
|
||||
|
||||
for channel in channels['channels']:
|
||||
dataMess = []
|
||||
dataUsers = []
|
||||
pprint( channel['name'] )
|
||||
begin = date - monthdelta(12)
|
||||
end = begin + monthdelta(1)
|
||||
|
||||
color = getColor()
|
||||
messageByChannel = {
|
||||
"label": channel['name'],
|
||||
"backgroundColor": color,
|
||||
"data": dataMess
|
||||
}
|
||||
|
||||
userByChannel = {
|
||||
"label": channel['name'],
|
||||
"backgroundColor": color,
|
||||
"data": dataUsers
|
||||
}
|
||||
tsunamy = getTsunamy(channel)
|
||||
|
||||
messagesByChannel.append(messageByChannel)
|
||||
usersByChannel.append(userByChannel)
|
||||
for id in range(0, 12):
|
||||
if usersTest[id] == None:
|
||||
usersTest[id] = []
|
||||
labels[id] = begin.strftime("%b %Y")
|
||||
begindate = begin.isoformat()
|
||||
enddate = end.isoformat()
|
||||
resultMess = rocket.channels_history(channel['_id'], oldest= begindate, latest=enddate, count= 10000).json()
|
||||
resultMess = list(filter(lambda mess: 't' not in mess, resultMess['messages']))
|
||||
lenght = len(resultMess)
|
||||
dataMess.append(lenght)
|
||||
|
||||
if channels['count'] + channels['offset'] >= channels['total']:
|
||||
break
|
||||
index += channels['count']
|
||||
if lenght > 0:
|
||||
if tsunamy & Tsunami.GLOBAL:
|
||||
messagesDataTsunamy['global'][id] += lenght
|
||||
if tsunamy & Tsunami.PROJECT:
|
||||
messagesDataTsunamy['project'][id] += lenght
|
||||
if tsunamy & Tsunami.DEMOCRACY:
|
||||
messagesDataTsunamy['democraty'][id] += lenght
|
||||
if tsunamy & Tsunami.ECOLOGY:
|
||||
messagesDataTsunamy['ecology'][id] += lenght
|
||||
if tsunamy & Tsunami.TECHNOLOGY:
|
||||
messagesDataTsunamy['technology'][id] += lenght
|
||||
|
||||
for id in range(0, 12):
|
||||
usersTest[id] = len(set(usersTest[id]))
|
||||
users = []
|
||||
for mess in resultMess:
|
||||
users.append(mess['u']['_id'])
|
||||
usersTest[id].append(mess['u']['_id'])
|
||||
usersDistinct = set(users)
|
||||
dataUsers.append(len(usersDistinct))
|
||||
else:
|
||||
dataUsers.append(0)
|
||||
|
||||
userGlobal = {
|
||||
"label": 'global',
|
||||
"backgroundColor": 'red',
|
||||
"data": usersTest
|
||||
}
|
||||
usersGlobal.append(userGlobal)
|
||||
begin = end
|
||||
end = begin + monthdelta(1)
|
||||
|
||||
color = getColor()
|
||||
messageByChannel = createElement(channel['name'], color,dataMess)
|
||||
userByChannel = createElement( channel['name'], color,dataUsers)
|
||||
|
||||
# Récupération du répertoire racine du repo
|
||||
rootFolder = os.path.join(os.path.dirname(__file__), '..')
|
||||
# Répertoire pour stocker le fichier de sortie
|
||||
dataFolder = os.path.join(rootFolder, 'public', 'data')
|
||||
messagesByChannel.append(messageByChannel)
|
||||
usersByChannel.append(userByChannel)
|
||||
|
||||
statsFilePath = os.path.abspath(
|
||||
os.path.join(dataFolder, 'channelsstat.json'))
|
||||
with open(statsFilePath, "w") as file_write:
|
||||
json.dump(info, file_write)
|
||||
if channels['count'] + channels['offset'] >= channels['total']:
|
||||
break
|
||||
index += channels['count']
|
||||
|
||||
for id in range(0, 12):
|
||||
usersTest[id] = len(set(usersTest[id]))
|
||||
|
||||
userGlobal = createElement( 'global', 'red', usersTest)
|
||||
usersGlobal.append(userGlobal)
|
||||
|
||||
save(info)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Reference in a new issue