def process_text(access_token, user_id, sticker_number, sticker_type, title,
urls, output_message_id):
bot = telegram.Bot(token=access_token)
sticker_name = get_sticker_name_from_sticker_number(
bot, sticker_type, sticker_number)
has_uploaded_first_sticker = False
backup_count = 0
new_sticker_name = sticker_name[:]
is_valid_sticker_number = False
while not is_valid_sticker_number:
print(new_sticker_name)
# check if there has been a sticker set
sticker_set = get_sticker_set(bot, new_sticker_name)
# three conditions:
# 1. no sticker set -> upload stickers
# 2. exist sticker set, and sticker set finished uploading -> return finished sticker set
# 3. exist sticker set, but sticker set didn't finish uploading -> delete old sticker set and upload stickers
if sticker_set != None:
# condition 2
if len(sticker_set.stickers) == len(urls):
bot.edit_message_text(
chat_id=user_id,
message_id=output_message_id,
text=(f"總算找到了\n"
f"This one?!\n\n"
f"Line sticker number:{sticker_number}"))
bot.send_sticker(
chat_id=user_id,
sticker=sticker_set.stickers[0].file_id,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(
text=title,
url=f"https://t.me/addstickers/{new_sticker_name}")
]]))
return
# condition 3
else:
delete_sticker_set(bot, sticker_set)
# upload
upload_static_text = (
f"{title}\n"
f"發現{len(urls)}張貼圖，轉換需要一段時間，完成時會再通知\n\n"
f"Found {len(urls)} stickers. You will be notified when finishing converting\n"
)
# first image for creating a sticker set
if not has_uploaded_first_sticker:
sticker_file = get_sticker_from_url(sticker_type, urls[0])
if sticker_type != "animated":
sticker_file.save(f"{sticker_number}.png")
sticker0 = bot.upload_sticker_file(user_id=user_id,
png_sticker=open(
f"{sticker_number}.png",
'rb')).file_id
else:
ffmpeg_write_video(sticker_file,
filename=f"{sticker_number}.webm",
fps=30,
codec="libvpx-vp9",
bitrate=None,
preset="medium",
with_mask=True,
write_logfile=False,
audiofile=None,
threads=None,
ffmpeg_params=[],
pixel_format=None)
sticker0 = f"{sticker_number}.webm"
has_uploaded_first_sticker = True
# create a sticker set
is_potential_valid_sticker_name = False
while not is_potential_valid_sticker_name:
try:
if sticker_type != "animated":
bot.create_new_sticker_set(
user_id=user_id,
name=new_sticker_name,
title=f"{title} @RekcitsEnilbot",
png_sticker=sticker0,
emojis=get_random_emoji())
else:
bot.create_new_sticker_set(
user_id=user_id,
name=new_sticker_name,
title=f"{title} @RekcitsEnilbot",
webm_sticker=open(sticker0, 'rb'),
emojis=get_random_emoji())
is_potential_valid_sticker_name = True
is_valid_sticker_number = True
upload_text = f"{upload_static_text}{'*' * 1}{'_' * (len(urls) - 1)}{1}/{len(urls)}"
bot.edit_message_text(chat_id=user_id,
message_id=output_message_id,
text=upload_text)
except BadRequest as e:
if str(e) == "Shortname_occupy_failed" or str(
e) == "Sticker set name is already occupied":
# A special error that I don't know what cause it.
# Telegram say that this is an internal error.....
new_sticker_name = f"backup_{backup_count}_{sticker_name}"
backup_count = backup_count + 1
is_potential_valid_sticker_name = True
else:
print("??????")
print(
'Error on line {}'.format(
sys.exc_info()[-1].tb_lineno),
type(e).__name__, e)
print(e)
return
sticker_name = new_sticker_name[:]
# the left images to be uploaded
for idx, url in enumerate(urls[1:]):
sticker_file = get_sticker_from_url(sticker_type, url)
if sticker_type != "animated":
sticker_file.save(f"{sticker_number}.png")
else:
ffmpeg_write_video(sticker_file,
filename=f"{sticker_number}.webm",
fps=30,
codec="libvpx-vp9",
bitrate=None,
preset="medium",
with_mask=True,
write_logfile=False,
audiofile=None,
threads=None,
ffmpeg_params=[],
pixel_format=None)
if sticker_type != "animated":
try:
sticker = bot.upload_sticker_file(user_id=user_id,
png_sticker=open(
f"{sticker_number}.png",
'rb')).file_id
except Exception as e:
w, h = sticker_image.size
print('Error on line {}'.format(sys.exc_info()[-1].tb_lineno),
type(e).__name__, e)
print(w, h)
print(url)
print(e)
return
bot.add_sticker_to_set(user_id=user_id,
name=sticker_name,
png_sticker=sticker,
emojis=get_random_emoji())
else:
bot.add_sticker_to_set(user_id=user_id,
name=sticker_name,
webm_sticker=open(f"{sticker_number}.webm",
'rb'),
emojis=get_random_emoji())
upload_text = f"{upload_static_text}{'*' * (idx + 2)}{'_' * (len(urls) - (idx + 2))}{idx + 2}/{len(urls)}"
bot.edit_message_text(chat_id=user_id,
message_id=output_message_id,
text=upload_text)
# delete temporary file
if sticker_type != "animated":
os.remove(f"{sticker_number}.png")
else:
os.remove(f"{sticker_number}.webm")
os.remove(f"temp.png")
os.remove(f"temp.gif")
# finish uploading
bot.send_message(chat_id=user_id,
text=(f"噠啦～☆\n\n"
f"Finished!\n\n"
f"Line sticker number:{sticker_number}\n"
f"https://t.me/addstickers/{sticker_name}"))
# send the first sticker of sticker set
sticker_set = get_sticker_set(bot, sticker_name)
bot.send_sticker(chat_id=user_id,
sticker=sticker_set.stickers[0].file_id,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(
text=title,
url=f"https://t.me/addstickers/{sticker_name}")
]]))
return
def write_videofile(
self,
filename,
fps=None,
codec=None,
bitrate=None,
audio=True,
audio_fps=44100,
preset="medium",
audio_nbytes=4,
audio_codec=None,
audio_bitrate=None,
audio_bufsize=2000,
temp_audiofile=None,
temp_audiofile_path="",
remove_temp=True,
write_logfile=False,
threads=None,
ffmpeg_params=None,
logger="bar",
pixel_format=None,
):
"""Write the clip to a videofile.
Parameters
----------
filename
Name of the video file to write in, as a string or a path-like object.
The extension must correspond to the "codec" used (see below),
or simply be '.avi' (which will work with any codec).
fps
Number of frames per second in the resulting video file. If None is
provided, and the clip has an fps attribute, this fps will be used.
codec
Codec to use for image encoding. Can be any codec supported
by ffmpeg. If the filename is has extension '.mp4', '.ogv', '.webm',
the codec will be set accordingly, but you can still set it if you
don't like the default. For other extensions, the output filename
must be set accordingly.
Some examples of codecs are:
- ``'libx264'`` (default codec for file extension ``.mp4``)
makes well-compressed videos (quality tunable using 'bitrate').
- ``'mpeg4'`` (other codec for extension ``.mp4``) can be an alternative
to ``'libx264'``, and produces higher quality videos by default.
- ``'rawvideo'`` (use file extension ``.avi``) will produce
a video of perfect quality, of possibly very huge size.
- ``png`` (use file extension ``.avi``) will produce a video
of perfect quality, of smaller size than with ``rawvideo``.
- ``'libvorbis'`` (use file extension ``.ogv``) is a nice video
format, which is completely free/ open source. However not
everyone has the codecs installed by default on their machine.
- ``'libvpx'`` (use file extension ``.webm``) is tiny a video
format well indicated for web videos (with HTML5). Open source.
audio
Either ``True``, ``False``, or a file name.
If ``True`` and the clip has an audio clip attached, this
audio clip will be incorporated as a soundtrack in the movie.
If ``audio`` is the name of an audio file, this audio file
will be incorporated as a soundtrack in the movie.
audio_fps
frame rate to use when generating the sound.
temp_audiofile
the name of the temporary audiofile, as a string or path-like object,
to be created and then used to write the complete video, if any.
temp_audiofile_path
the location that the temporary audiofile is placed, as a
string or path-like object. Defaults to the current working directory.
audio_codec
Which audio codec should be used. Examples are 'libmp3lame'
for '.mp3', 'libvorbis' for 'ogg', 'libfdk_aac':'m4a',
'pcm_s16le' for 16-bit wav and 'pcm_s32le' for 32-bit wav.
Default is 'libmp3lame', unless the video extension is 'ogv'
or 'webm', at which case the default is 'libvorbis'.
audio_bitrate
Audio bitrate, given as a string like '50k', '500k', '3000k'.
Will determine the size/quality of audio in the output file.
Note that it mainly an indicative goal, the bitrate won't
necessarily be the this in the final file.
preset
Sets the time that FFMPEG will spend optimizing the compression.
Choices are: ultrafast, superfast, veryfast, faster, fast, medium,
slow, slower, veryslow, placebo. Note that this does not impact
the quality of the video, only the size of the video file. So
choose ultrafast when you are in a hurry and file size does not
matter.
threads
Number of threads to use for ffmpeg. Can speed up the writing of
the video on multicore computers.
ffmpeg_params
Any additional ffmpeg parameters you would like to pass, as a list
of terms, like ['-option1', 'value1', '-option2', 'value2'].
write_logfile
If true, will write log files for the audio and the video.
These will be files ending with '.log' with the name of the
output file in them.
logger
Either ``"bar"`` for progress bar or ``None`` or any Proglog logger.
pixel_format
Pixel format for the output video file.
Examples
--------
>>> from moviepy import VideoFileClip
>>> clip = VideoFileClip("myvideo.mp4").subclip(100,120)
>>> clip.write_videofile("my_new_video.mp4")
>>> clip.close()
"""
name, ext = os.path.splitext(os.path.basename(filename))
ext = ext[1:].lower()
logger = proglog.default_bar_logger(logger)
if codec is None:
try:
codec = extensions_dict[ext]["codec"][0]
except KeyError:
raise ValueError("MoviePy couldn't find the codec associated "
"with the filename. Provide the 'codec' "
"parameter in write_videofile.")
if audio_codec is None:
if ext in ["ogv", "webm"]:
audio_codec = "libvorbis"
else:
audio_codec = "libmp3lame"
elif audio_codec == "raw16":
audio_codec = "pcm_s16le"
elif audio_codec == "raw32":
audio_codec = "pcm_s32le"
audiofile = audio if isinstance(audio, str) else None
make_audio = ((audiofile is None) and (audio is True)
and (self.audio is not None))
if make_audio and temp_audiofile:
# The audio will be the clip's audio
audiofile = temp_audiofile
elif make_audio:
audio_ext = find_extension(audio_codec)
audiofile = os.path.join(
temp_audiofile_path,
name + Clip._TEMP_FILES_PREFIX + "wvf_snd.%s" % audio_ext,
)
# enough cpu for multiprocessing ? USELESS RIGHT NOW, WILL COME AGAIN
# enough_cpu = (multiprocessing.cpu_count() > 1)
logger(message="Moviepy - Building video %s." % filename)
if make_audio:
self.audio.write_audiofile(
audiofile,
audio_fps,
audio_nbytes,
audio_bufsize,
audio_codec,
bitrate=audio_bitrate,
write_logfile=write_logfile,
logger=logger,
)
ffmpeg_write_video(
self,
filename,
fps,
codec,
bitrate=bitrate,
preset=preset,
write_logfile=write_logfile,
audiofile=audiofile,
threads=threads,
ffmpeg_params=ffmpeg_params,
logger=logger,
pixel_format=pixel_format,
)
if remove_temp and make_audio:
if os.path.exists(audiofile):
os.remove(audiofile)
logger(message="Moviepy - video ready %s" % filename)
def test_ffmpeg_write_video(
util,
codec,
is_valid_codec,
ext,
write_logfile,
with_mask,
bitrate,
threads,
):
filename = os.path.join(util.TMP_DIR, f"moviepy_ffmpeg_write_video{ext}")
if os.path.isfile(filename):
try:
os.remove(filename)
except PermissionError:
pass
logfile_name = filename + ".log"
if os.path.isfile(logfile_name):
os.remove(logfile_name)
clip = BitmapClip([["R"], ["G"], ["B"]], fps=10).with_duration(0.3)
if with_mask:
clip = clip.with_mask(
BitmapClip([["W"], ["O"], ["O"]], fps=10,
is_mask=True).with_duration(0.3))
kwargs = dict(
logger=None,
write_logfile=write_logfile,
with_mask=with_mask,
)
if codec is not None:
kwargs["codec"] = codec
if bitrate is not None:
kwargs["bitrate"] = bitrate
if threads is not None:
kwargs["threads"] = threads
ffmpeg_write_video(clip, filename, 10, **kwargs)
if is_valid_codec:
assert os.path.isfile(filename)
final_clip = VideoFileClip(filename)
r, g, b = final_clip.get_frame(0)[0][0]
assert r == 254
assert g == 0
assert b == 0
r, g, b = final_clip.get_frame(0.1)[0][0]
assert r == (0 if not with_mask else 1)
assert g == (255 if not with_mask else 1)
assert b == 1
r, g, b = final_clip.get_frame(0.2)[0][0]
assert r == 0
assert g == 0
assert b == (255 if not with_mask else 0)
if write_logfile:
assert os.path.isfile(logfile_name)
