generated from pavel.muhortov/template-bash
add Do.tg_routine_media()
This commit is contained in:
parent
aea4a5ba52
commit
c575e844f2
|
|
@ -12,7 +12,7 @@ from argparse import ArgumentParser
|
|||
from datetime import datetime
|
||||
from ftplib import FTP
|
||||
from multiprocessing import Process, Queue
|
||||
from os import environ, makedirs, path, remove, replace, sep, walk
|
||||
from os import environ, makedirs, path, remove, replace, sep, stat, walk
|
||||
from subprocess import Popen, PIPE, STDOUT
|
||||
from sys import platform
|
||||
from time import sleep
|
||||
|
|
@ -2393,12 +2393,24 @@ class Do():
|
|||
if Do.args_valid(locals(), Do.wp_routine_media.__annotations__):
|
||||
default_media_links = {
|
||||
"d": {
|
||||
"point-01": "https://www.hmp.today/wp-content/uploads/2022/07/point-01_yyyy.mm_.dd_.mp4",
|
||||
"point-02": "https://www.hmp.today/wp-content/uploads/2022/07/point-02_yyyy.mm_.dd_.mp4",
|
||||
"point-04": "https://www.hmp.today/wp-content/uploads/2022/07/point-04_yyyy.mm_.dd_.mp4",
|
||||
"point-05": "https://www.hmp.today/wp-content/uploads/2022/07/point-05_yyyy.mm_.dd_.mp4",
|
||||
"point-11": "https://www.hmp.today/wp-content/uploads/2022/07/point-11_yyyy.mm_.dd_.mp4",
|
||||
"point-12": "https://www.hmp.today/wp-content/uploads/2022/07/point-12_yyyy.mm_.dd_.mp4"
|
||||
"point-01": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-01_yyyy.mm_.dd_.mp4"
|
||||
),
|
||||
"point-02": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-02_yyyy.mm_.dd_.mp4"
|
||||
),
|
||||
"point-04": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-04_yyyy.mm_.dd_.mp4"
|
||||
),
|
||||
"point-05": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-05_yyyy.mm_.dd_.mp4"
|
||||
),
|
||||
"point-11": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-11_yyyy.mm_.dd_.mp4"
|
||||
),
|
||||
"point-12": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-12_yyyy.mm_.dd_.mp4"
|
||||
)
|
||||
},
|
||||
"w": {
|
||||
"point-01": "https://www.hmp.today/wp-content/uploads/2022/07/point-01_w.mp4",
|
||||
|
|
@ -2409,20 +2421,44 @@ class Do():
|
|||
"point-12": "https://www.hmp.today/wp-content/uploads/2022/07/point-12_w.mp4"
|
||||
},
|
||||
"m": {
|
||||
"point-01": "https://www.hmp.today/wp-content/uploads/2022/07/point-01_yyyy.mm_.mp4",
|
||||
"point-02": "https://www.hmp.today/wp-content/uploads/2022/07/point-02_yyyy.mm_.mp4",
|
||||
"point-04": "https://www.hmp.today/wp-content/uploads/2022/07/point-04_yyyy.mm_.mp4",
|
||||
"point-05": "https://www.hmp.today/wp-content/uploads/2022/07/point-05_yyyy.mm_.mp4",
|
||||
"point-11": "https://www.hmp.today/wp-content/uploads/2022/07/point-11_yyyy.mm_.mp4",
|
||||
"point-12": "https://www.hmp.today/wp-content/uploads/2022/07/point-12_yyyy.mm_.mp4"
|
||||
"point-01": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-01_yyyy.mm_.mp4"
|
||||
),
|
||||
"point-02": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-02_yyyy.mm_.mp4"
|
||||
),
|
||||
"point-04": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-04_yyyy.mm_.mp4"
|
||||
),
|
||||
"point-05": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-05_yyyy.mm_.mp4"
|
||||
),
|
||||
"point-11": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-11_yyyy.mm_.mp4"
|
||||
),
|
||||
"point-12": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-12_yyyy.mm_.mp4"
|
||||
)
|
||||
},
|
||||
"y": {
|
||||
"point-01": "https://www.hmp.today/wp-content/uploads/2022/07/point-01_yyyy.mp4",
|
||||
"point-02": "https://www.hmp.today/wp-content/uploads/2022/07/point-02_yyyy.mp4",
|
||||
"point-04": "https://www.hmp.today/wp-content/uploads/2022/07/point-04_yyyy.mp4",
|
||||
"point-05": "https://www.hmp.today/wp-content/uploads/2022/07/point-05_yyyy.mp4",
|
||||
"point-11": "https://www.hmp.today/wp-content/uploads/2022/07/point-11_yyyy.mp4",
|
||||
"point-12": "https://www.hmp.today/wp-content/uploads/2022/07/point-12_yyyy.mp4"
|
||||
"point-01": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-01_yyyy.mp4"
|
||||
),
|
||||
"point-02": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-02_yyyy.mp4"
|
||||
),
|
||||
"point-04": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-04_yyyy.mp4"
|
||||
),
|
||||
"point-05": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-05_yyyy.mp4"
|
||||
),
|
||||
"point-11": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-11_yyyy.mp4"
|
||||
),
|
||||
"point-12": (
|
||||
"https://www.hmp.today/wp-content/uploads/2022/07/point-12_yyyy.mp4"
|
||||
)
|
||||
}
|
||||
}
|
||||
current_media_links = default_media_links[period]
|
||||
|
|
@ -2595,6 +2631,147 @@ class Do():
|
|||
|
||||
return result
|
||||
|
||||
@staticmethod
|
||||
# pylint: disable=W0612,W0511
|
||||
def tg_routine_media(
|
||||
tg: Telegram,
|
||||
targets_media_files: dict,
|
||||
period: str,
|
||||
amount: int,
|
||||
chat: str
|
||||
) -> dict:
|
||||
"""Custom Telegram media routine - send mediagroup.
|
||||
|
||||
Args:
|
||||
tg (Telegram): telegram object
|
||||
targets_media_files (dict): {'period':{'name':'local/path/to/file'}}.
|
||||
period (str): 'y','m','w','d'.
|
||||
amount (int): +/- periods.
|
||||
chat (str): unique identifier for the target chat or username of the target channel.
|
||||
|
||||
Raises:
|
||||
ValueError: filename is not local file.
|
||||
|
||||
Returns:
|
||||
dict: {'success':bool,'result':response}.
|
||||
"""
|
||||
if Do.args_valid(locals(), Do.tg_routine_media.__annotations__):
|
||||
default_caption = (""
|
||||
+ "`period:` yyyy.mm.dd\n"
|
||||
+ "`source:` https://www.hmp.today/media\n"
|
||||
+ "`stream:` https://youtu.be/ikB20ML5oyo"
|
||||
)
|
||||
group = {
|
||||
"cover":
|
||||
{
|
||||
"type": "photo",
|
||||
"path": (''
|
||||
+ 'https://www.hmp.today/wp-content/uploads/'
|
||||
+ '2021/02/site-slider_hmp-qr_bwg-3840x1705-1.png'
|
||||
),
|
||||
"caption": "source: https://www.hmp.today"
|
||||
}
|
||||
}
|
||||
tmp_files = []
|
||||
|
||||
date = Do.date_calc(period=period, amount=amount)
|
||||
if period == 'd':
|
||||
caption_date = (''
|
||||
+ str(date['start']['y'])
|
||||
+ '.'
|
||||
+ str(date['start']['m']).zfill(2)
|
||||
+ '.'
|
||||
+ str(date['start']['d']).zfill(2)
|
||||
)
|
||||
if period == 'w':
|
||||
caption_date = (''
|
||||
+ str(date['start']['y'])
|
||||
+ '-w'
|
||||
+ str(date['start']['w']).zfill(2)
|
||||
)
|
||||
if period == 'm':
|
||||
caption_date = (''
|
||||
+ str(date['start']['y'])
|
||||
+ '.'
|
||||
+ str(date['start']['m']).zfill(2)
|
||||
)
|
||||
if period == 'y':
|
||||
caption_date = (''
|
||||
+ str(date['start']['y'])
|
||||
)
|
||||
current_caption = default_caption.replace('yyyy.mm.dd', caption_date)
|
||||
|
||||
for media_name, media_path in targets_media_files[period].items():
|
||||
if re.match("^(?:http://|https://|file_id=)", media_path):
|
||||
# todo: check remote file size
|
||||
raise ValueError(media_name + ' is not local file')
|
||||
else:
|
||||
tg_limit_video_size = 50000000
|
||||
src_video_info = FFmpeg.probe(target=media_path)
|
||||
if src_video_info:
|
||||
cur_video_file = media_path
|
||||
cur_video_duration = int(float(src_video_info['format']['duration']))
|
||||
cur_video_fps = int(
|
||||
src_video_info['streams'][0]['avg_frame_rate'].split('/')[0]
|
||||
)
|
||||
cur_video_width = int(src_video_info['streams'][0]['width'])
|
||||
cur_video_height = int(src_video_info['streams'][0]['height'])
|
||||
|
||||
if stat(media_path).st_size >= tg_limit_video_size:
|
||||
tmp_video_bitrate = int(
|
||||
tg_limit_video_size*0.9/1000*8/cur_video_duration
|
||||
)
|
||||
tmp_video_file = media_path.replace('.mp4', '_compressed.mp4')
|
||||
cur_video_width = int(cur_video_width/2)
|
||||
cur_video_height = int(cur_video_height/2)
|
||||
|
||||
compressing_params = ' '.join(
|
||||
[
|
||||
'-i', media_path,
|
||||
'-c:v libx264 -b:v', str(tmp_video_bitrate) + 'k',
|
||||
'-vf scale='
|
||||
+ str(cur_video_width) + ':' + str(cur_video_height)
|
||||
+ ',fps=' + str(cur_video_fps)
|
||||
+ ',format=yuv420p -preset veryslow',
|
||||
tmp_video_file, '-y -loglevel quiet -stats'
|
||||
]
|
||||
)
|
||||
if FFmpeg.run(raw=compressing_params) == 0:
|
||||
logging.info(msg=media_path + " comressed to " + tmp_video_file)
|
||||
tmp_files.append(tmp_video_file)
|
||||
cur_video_file = tmp_video_file
|
||||
|
||||
response_upload = tg.send_video(
|
||||
chat=chat,
|
||||
video=cur_video_file,
|
||||
width=cur_video_width,
|
||||
height=cur_video_height,
|
||||
duration=cur_video_duration
|
||||
)
|
||||
if response_upload:
|
||||
response_delete = tg.delete_message(
|
||||
chat=chat,
|
||||
message_id=response_upload['result']['result']['message_id']
|
||||
)
|
||||
group[media_name] = {
|
||||
"type": "video",
|
||||
"path": (
|
||||
'file_id='
|
||||
+ response_upload['result']['result']['video']['file_id']
|
||||
)
|
||||
}
|
||||
|
||||
response_result = tg.send_mediagroup(
|
||||
chat=chat,
|
||||
media=group,
|
||||
caption=current_caption,
|
||||
parse_mode='Markdown'
|
||||
)
|
||||
for file in tmp_files:
|
||||
if remove(file):
|
||||
logging.info(msg="deleted " + file)
|
||||
return response_result
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
time_start = datetime.now()
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user