diff --git a/snapchat_bots/utils.py b/snapchat_bots/utils.py index 1425804..a172908 100755 --- a/snapchat_bots/utils.py +++ b/snapchat_bots/utils.py @@ -1,4 +1,4 @@ -import tempfile, mimetypes, datetime, subprocess, re, math +import tempfile, mimetypes, datetime, subprocess, re, math, os from PIL import Image from constants import MEDIA_TYPE_IMAGE, MEDIA_TYPE_VIDEO, MEDIA_TYPE_VIDEO_WITHOUT_AUDIO, SNAP_IMAGE_DIMENSIONS @@ -11,14 +11,19 @@ def file_extension_for_type(media_type): def create_temporary_file(suffix): return tempfile.NamedTemporaryFile(suffix=suffix, delete=False) -def save_snap(snap): +def save_snap(snap,dir_name): now = datetime.datetime.now() + before_folder = os.getcwd() + if not os.path.exists(dir_name): + os.makedirs(dir_name) + os.chdir(dir_name) filename = '%s-%s.%s.%s-%s:%s:%s%s' % (snap.sender, now.month, now.day, now.year, now.hour, now.minute, now.second, snap.file.name[-4:]) with open(filename, 'wb') as f: data = snap.file.file.read(8192) while data: f.write(data) data = snap.file.file.read(8192) + os.chdir(before_folder) def is_video_file(path): return mimetypes.guess_type(path)[0].startswith("video")