-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
119 lines (101 loc) · 3.82 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from flask import Flask, request, jsonify, send_from_directory
import instaloader
import re
import os
import shutil
app = Flask(__name__)
# Create an instance of Instaloader with a session context
L = instaloader.Instaloader()
# Function to extract shortcode from URL
def extract_shortcode(url):
pattern = re.compile(r"(?:https?://)?(?:www\.)?instagram\.com/(?:p|reel)/([^/?#&]+)")
match = pattern.search(url)
if match:
return match.group(1)
return url
# Function to download post using Instaloader
def download_instagram_post(shortcode):
try:
# Fetch post by shortcode
post = instaloader.Post.from_shortcode(L.context, shortcode)
# Create a directory for the shortcode if it doesn't exist
target_dir = os.path.join(os.getcwd(), shortcode)
os.makedirs(target_dir, exist_ok=True)
# Change the download directory for Instaloader
L.dirname_pattern = target_dir
# Download post to the target directory
L.download_post(post, target=shortcode)
# Retrieve all filenames in the shortcode folder
file_urls = []
for filename in os.listdir(target_dir):
if filename.endswith(('.jpg', '.mp4')): # Filtering images and videos
download_url = f"http://{request.host}/{shortcode}/{filename}"
file_urls.append(download_url)
result = {
'status': 'success',
'shortcode': shortcode,
'download_urls': file_urls
}
except instaloader.exceptions.InstaloaderException as e:
result = {
'status': 'failed',
'error': str(e)
}
except Exception as e:
result = {
'status': 'failed',
'error': f"Failed to fetch post data: {str(e)}"
}
return result
@app.route('/download', methods=['POST'])
def handle_download():
data = request.json
post_url = data.get('url')
if post_url:
shortcode = extract_shortcode(post_url)
download_result = download_instagram_post(shortcode)
download_result['url'] = post_url
else:
download_result = {
'status': 'failed',
'error': 'No URL provided in request data'
}
return jsonify(download_result)
@app.route('/<shortcode>/<filename>', methods=['GET'])
def serve_file(shortcode, filename):
directory = os.path.join(os.getcwd(), shortcode)
return send_from_directory(directory, filename)
@app.route('/remove_folder', methods=['POST'])
def remove_folder():
data = request.json
shortcode = data.get('shortcode')
if shortcode:
try:
# Construct path to folder using shortcode
folder_path = os.path.join(os.getcwd(), shortcode)
# Check if folder exists
if os.path.exists(folder_path):
# Remove the folder and its contents
shutil.rmtree(folder_path)
result = {
'status': 'success',
'message': f"Folder with shortcode {shortcode} removed successfully"
}
else:
result = {
'status': 'failed',
'error': f"Folder with shortcode {shortcode} does not exist"
}
except Exception as e:
result = {
'status': 'failed',
'error': f"Error removing folder: {str(e)}"
}
else:
result = {
'status': 'failed',
'error': 'No shortcode provided in request data'
}
return jsonify(result)
if __name__ == '__main__':
app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 5000)))