Add files via upload

src/analyse_and_bot_functions.py

@@ -8,13 +8,25 @@
 import time
 import random
 import math
+import matplotlib.pyplot as plt
+from matplotlib.ticker import AutoMinorLocator
+import pylab as pltt
+import datetime
+import matplotlib.dates as mdates
+
+
 
 
 token = "65e6efa565e6efa565e6efa54f6593fb1f665e665e6efa53a5c6937a4636b3416a8bd92"
 group_token = "17e681fbe171945431a04f1abc752d41ff888698288abf74124de4e782c67f36e76484601991870f56b7a"
+new_token = "812c2975fc2ac0785252d97e8b5011f45e873a00dfb98b15299aec060ff7b890d06c4822feab0626e198c"
+our_group_id = 200698416
+
 analyse_group_id = 'memkn'
 my_group_id = 'memkn_funclub'
 my_number_group_id = 200698416
+album_id = 278041850
+version = '5.95'
 
 session1 = vk.AuthSession(access_token=token)
 session2 = vk.AuthSession(access_token=group_token)
@@ -25,6 +37,29 @@
 days_of_the_week = ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"]
 
 
+def upload_picture(picture_path):
+    """
+    uploading a picture to an album in vk group
+    :param picture_path:
+    :return an answer of request:
+    """
+    r = vk_api.photos.getUploadServer(group_id=our_group_id, album_id=album_id)
+    url = r['upload_url']
+    file = {'file1': open(picture_path, 'rb')}
+    ur = requests.post(url, files=file).json()
+    result = requests.get('https://api.vk.com/method/photos.save',
+                          params={
+                              'access_token': new_token,
+                              'album_id': ur['aid'],
+                              'group_id': ur['gid'],
+                              'server': ur['server'],
+                              'photos_list': ur['photos_list'],
+                              'hash': ur['hash'],
+                              'v': version,
+                          }).json()
+    return url
+
+
 def get_message(group_id, server_, ts_, key_):
     # gets the message from the user
     response = requests.get('{server}?act=a_check&key={key}&ts={ts}&wait=25'.format
@@ -52,7 +87,6 @@ def process_input_message(message):
     code 10: the user needs the advice how to give a task
     code 11: user wants to set time and has pressed the button
     code 12: user wants to set recommendation time and has entered this time already
-
     :param message: the message that was sent by the user
     :return: special code, string - the group id or "" - and a number - the period of analysing or -1
     """
@@ -366,10 +400,91 @@ def not_available(current_user_id):
     return
 
 
+# dicts belong -- samples of dicts for functions
+
+
+def time_from_db_to_date(time_string):
+    """
+    time in table is string, so this formatting it to a timedate. example:
+    time_string = "Mon, 10:30"
+                   0123456789
+    :param time_string:
+    :return timedate:
+    """
+
+    moment_hours = int(time_string[5])*10 + int(time_string[6])
+    moment_minutes = int(time_string[8])*10 + int(time_string[9])
+    moment_days = 7
+    if time_string[:3] == "Mon":
+        moment_days = 1
+    if time_string[:3] == "Tue":
+        moment_days = 2
+    if time_string[:3] == "Wed":
+        moment_days = 3
+    if time_string[:3] == "Thu":
+        moment_days = 4
+    if time_string[:3] == "Fri":
+        moment_days = 5
+    if time_string[:3] == "Sat":
+        moment_days = 6
+    auxiliary_delta = datetime.timedelta(days=moment_days, hours=moment_hours, minutes=moment_minutes)
+    # time from the beginning of the week
+    abstract_sunday = datetime.datetime(2020, 12, 27)
+    final_time = abstract_sunday + auxiliary_delta
+    return final_time
+
+
+def dict_with_strings_to_dict_for_plots(dict_with_strings):
+    """
+    converting result of request to db to dict for plot funstions
+    :param dict_with_strings:
+    :return:
+    """
+    new_dict = {}
+    for key in dict_with_strings:
+        key_for_new_dict = time_from_db_to_date(key)
+        new_dict[key_for_new_dict] = dict_with_strings[key]
+    return new_dict
+
+
+def create_daily_image(dict_with_data, label_of_image):
+    """
+    an image for daily report
+    :param dict_with_data:
+    :param label_of_image:
+    :return nothing:
+    """
+    day_delta = 0
+    prev_key = datetime.datetime.now()
+    for key in dict_with_data:
+        day_delta = key - prev_key
+        prev_key = key
+    period = day_delta.total_seconds()
+    period = period // 60
+    number_of_dots = int(1440 // period)
+    y_axis = [0] * int(number_of_dots)
+    x_axis = [datetime.datetime(2020, 1, 1, 0, 0, 0) + day_delta * i for i in range(number_of_dots)]
+    for key in dict_with_data:
+        y_axis[int((key.minute + key.hour * 60) // period)] = dict_with_data[key]
+    # belong is decoration for graph
+    figure, ax = plt.subplots(figsize=(number_of_dots, 10))
+    ax.set_title(label_of_image)
+    ax.set_xlabel("Время", fontsize=14)
+    ax.set_ylabel("Процент онлайна", fontsize=14)
+    ax.grid(which="major", linewidth=1.2)
+    ax.grid(which="minor", linestyle="--", color="gray", linewidth=0.5)
+    ax.scatter(x_axis, y_axis, c="red")
+    ax.plot(x_axis, y_axis)
+    my_fmt = mdates.DateFormatter('%H:%M')
+    ax.xaxis.set_major_formatter(my_fmt)
+    figure.savefig("../data/images/" + label_of_image + ".png")
+    return
+
+
 def task_by_button(current_user_id):
     r_id = get_new_random_id()
     vk_api2.messages.send(
         user_id=current_user_id,
         message="Send the group_id and the period with a whitespace between them and '$' in the beginning",
         random_id=r_id
-    )
+    )

src/new_main.py

@@ -86,9 +86,10 @@
             group.recommendation_for_this_week()
         next_recommend += datetime.timedelta(days=1)
 
-    code, return_message, frequency_number = func.process_input_message(message)
+    code, return_message, return_number = func.process_input_message(message)
 
     if code == 3:
+        frequency_number = return_number
         analyse_group_id = return_message
         # this code means that user gave a correct task
         if not have_a_task or current_user_id == master_id:
@@ -172,3 +173,19 @@
             file.close()
         else:
             not func.not_available(current_user_id)
+    elif code == 13:
+        if have_a_task and current_user_id == master_id:
+            stats_dict = group.daily_graph_request(return_number)
+            stats_dict = func.dict_with_strings_to_dict_for_plots(stats_dict)
+            ph_id = "photos" + str(current_user_id) + \
+                    datetime.datetime.now().day + datetime.datetime.now().hour + \
+                    datetime.datetime.now().minute + datetime.datetime.now().second
+            func.create_daily_image(stats_dict, ph_id)
+            res_url = func.upload_picture(f'../data/image/{ph_id}')
+            stop_ind = res_url.find('%')
+            vk_api2.messages.send(
+                user_id=current_user_id,
+                message=f"Get it!",
+                random_id=r_id,
+                attachment=res_url[32: stop_ind]
+            )