Antonn/utils.py at master · sayyss/Antonn · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import datetime
import matplotlib.pyplot as plt
import numpy as np
import random
import string

validActivities = ['ActivityType.playing','ActivityType.watching','ActivityType.streaming','ActivityType.listening','ActivityType.custom']

def getAvgMessage(time,totalMsg):

    timeAdded = datetime.datetime.fromtimestamp(time)
    currentTime = datetime.date.today()

    timeAdded = datetime.date(timeAdded.year,timeAdded.month,timeAdded.day)
    currentTime = datetime.date(currentTime.year,currentTime.month,currentTime.day)

    days = currentTime - timeAdded

    if days.days == 0:
        return totalMsg

    else:
        return totalMsg/days.days

def getDaysSinceAdded(time):

    timeAdded = datetime.datetime.fromtimestamp(time)
    currentTime = datetime.date.today()

    timeAdded = datetime.date(timeAdded.year,timeAdded.month,timeAdded.day)
    currentTime = datetime.date(currentTime.year,currentTime.month,currentTime.day)

    days = currentTime - timeAdded

    return days.days

def getPlot(data):

    x = []
    y = []

    for i in data:

        time = i['time']
        time = datetime.datetime.fromtimestamp(time)
        time = time.date()
        time = str(time)

        x.append(time)
        y.append(i['count'])

    return x,y


def timestamp_to_date(time):

    date = datetime.datetime.fromtimestamp(time)
    date = date.date()
    date = str(date)

    return date

def generatePass():

    letters_and_digits = string.ascii_letters + string.digits
    result_str = ''.join((random.choice(letters_and_digits) for i in range(20)))

    return result_str

def getGuildChannels(channels):
    guildChannels = []

    for i in channels:
        channel = {
            "name": i.name,
            "id": i.id,
            "total_msg": 0
        }

        guildChannels.append(channel)

    return guildChannels

def getGuildMembers(members):

    guildMembers = []

    for i in members:

        if i.bot:
            continue

        member = {
            "id": i.id,
            "name": i.display_name,
            "total_msg": 0,
            'voice': 0,
            'voiceState': {'joined': None, 'left': None}
        }

        guildMembers.append(member)

    return guildMembers
def checkActivitiy(activity):

    exist = True if str(activity) in validActivities else False
    return exist