forked from UsergeTeam/Userge
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
update webss(add chrome driver) and memes(reacts, hmm, lol, ...) + ca…
…rbon + autopic(DB supported)
- Loading branch information
Showing
20 changed files
with
480 additions
and
223 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -140,3 +140,6 @@ gen | |
| unknown_errors.txt | ||
| logs/ | ||
| bin/ | ||
| resources/base_profile_pic.jpg | ||
| resources/mdfy_profile_pic.jpg | ||
| pictest.py | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Binary file not shown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| # Copyright (C) 2020 by UsergeTeam@Github, < https://github.com/UsergeTeam >. | ||
| # | ||
| # This file is part of < https://github.com/UsergeTeam/Userge > project, | ||
| # and is released under the "GNU v3.0 License Agreement". | ||
| # Please see < https://github.com/uaudith/Userge/blob/master/LICENSE > | ||
| # | ||
| # All rights reserved. | ||
|
|
||
|
|
||
| import os | ||
| import base64 | ||
| import asyncio | ||
| import datetime | ||
| import textwrap | ||
| from shutil import copyfile | ||
|
|
||
| import aiofiles | ||
| from PIL import Image, ImageFont, ImageDraw | ||
|
|
||
| from userge import userge, Message, Config, get_collection | ||
|
|
||
| SAVED_SETTINGS = get_collection("CONFIGS") | ||
|
|
||
| __tmp__ = SAVED_SETTINGS.find_one({'_id': 'UPDATE_PIC'}) | ||
|
|
||
| UPDATE_PIC = False | ||
| BASE_PIC = "resources/base_profile_pic.jpg" | ||
| MDFY_PIC = "resources/mdfy_profile_pic.jpg" | ||
| if __tmp__: | ||
| UPDATE_PIC = __tmp__['on'] | ||
| if not os.path.exists(BASE_PIC): | ||
| with open(BASE_PIC, "wb") as media_file_: | ||
| media_file_.write(base64.b64decode(__tmp__['media'])) | ||
|
|
||
| del __tmp__ | ||
|
|
||
| LOG = userge.getLogger(__name__) | ||
|
|
||
|
|
||
| @userge.on_cmd("autopic", about={ | ||
| 'header': "set profile picture", | ||
| 'usage': "{tr}autopic\n{tr}autopic [image path]\nset timeout using {tr}sappto"}) | ||
| async def autopic(message: Message): | ||
| global UPDATE_PIC | ||
| await message.edit('`processing...`') | ||
| if UPDATE_PIC: | ||
| if isinstance(UPDATE_PIC, asyncio.Task): | ||
| UPDATE_PIC.cancel() | ||
| UPDATE_PIC = False | ||
| SAVED_SETTINGS.update_one({'_id': 'UPDATE_PIC'}, | ||
| {"$set": {'on': False}}, upsert=True) | ||
| await message.edit('auto profile picture updation has been **stopped**', | ||
| del_in=5, log=__name__) | ||
| return | ||
| image_path = message.input_str | ||
| store = False | ||
| if os.path.exists(BASE_PIC) and not image_path: | ||
| pass | ||
| elif not image_path: | ||
| profile_photo = await userge.get_profile_photos("me", limit=1) | ||
| if not profile_photo: | ||
| await message.err("sorry, couldn't find any picture!") | ||
| return | ||
| await userge.download_media(profile_photo[0], file_name=BASE_PIC) | ||
| store = True | ||
| else: | ||
| if not os.path.exists(image_path): | ||
| await message.err("input path not found!") | ||
| return | ||
| if os.path.exists(BASE_PIC): | ||
| os.remove(BASE_PIC) | ||
| copyfile(image_path, BASE_PIC) | ||
| store = True | ||
| data_dict = {'on': True} | ||
| if store: | ||
| async with aiofiles.open(BASE_PIC, "rb") as media_file: | ||
| media = base64.b64encode(await media_file.read()) | ||
| data_dict['media'] = media | ||
| SAVED_SETTINGS.update_one({'_id': 'UPDATE_PIC'}, | ||
| {"$set": data_dict}, upsert=True) | ||
| await message.edit( | ||
| 'auto profile picture updation has been **started**', del_in=3, log=__name__) | ||
| UPDATE_PIC = asyncio.create_task(apic_worker()) | ||
|
|
||
|
|
||
| @userge.add_task | ||
| async def apic_worker(): | ||
| user_dict = await userge.get_user_dict('me') | ||
| user = '@' + user_dict['uname'] if user_dict['uname'] else user_dict['flname'] | ||
| while UPDATE_PIC: | ||
| img = Image.open(BASE_PIC) | ||
| i_width, i_height = img.size | ||
| s_font = ImageFont.truetype("resources/font.ttf", int((35 / 640)*i_width)) | ||
| l_font = ImageFont.truetype("resources/font.ttf", int((50 / 640)*i_width)) | ||
| draw = ImageDraw.Draw(img) | ||
| current_h, pad = 10, 0 | ||
| for user in textwrap.wrap(user, width=20): | ||
| u_width, u_height = draw.textsize(user, font=l_font) | ||
| draw.text(xy=((i_width - u_width) / 2, int((current_h / 640)*i_width)), text=user, | ||
| font=l_font, fill=(255, 255, 255)) | ||
| current_h += u_height + pad | ||
| tim = datetime.datetime.now( | ||
| tz=datetime.timezone(datetime.timedelta(minutes=30, hours=5))) | ||
| date_time = (f"DATE: {tim.day}.{tim.month}.{tim.year}\n" | ||
| f"TIME: {tim.hour}:{tim.minute}:{tim.second}\n" | ||
| "UTC+5:30") | ||
| d_width, d_height = draw.textsize(date_time, font=s_font) | ||
| draw.multiline_text( | ||
| xy=((i_width - d_width) / 2, i_height - d_height - int((20 / 640)*i_width)), | ||
| text=date_time, fill=(255, 255, 255), font=s_font, align="center") | ||
| img.convert('RGB').save(MDFY_PIC) | ||
| await userge.set_profile_photo(MDFY_PIC) | ||
| os.remove(MDFY_PIC) | ||
| LOG.info("profile photo has been updated!") | ||
| await asyncio.sleep(Config.AUTOPIC_TIMEOUT) |
Oops, something went wrong.