Spaces:
Sleeping
Sleeping
admin
commited on
Commit
·
9d75477
1
Parent(s):
022aaa4
update
Browse files- src/modules/bitly-api.min.py +10 -6
src/modules/bitly-api.min.py
CHANGED
@@ -79,9 +79,9 @@ class ShortenUrl:
|
|
79 |
buffered = BytesIO()
|
80 |
img.save(buffered, format="PNG")
|
81 |
qr_code_png = base64.b64encode(buffered.getvalue())
|
82 |
-
img_str = qr_code_png.decode()
|
83 |
|
84 |
-
qr_code_html = f'<img src="data:image/png;base64,{img_str}" alt="QR Code" height="40px">'
|
85 |
json_str = {"url": url}
|
86 |
return qr_code_html, buffered.getvalue(), json_str
|
87 |
except Exception as e:
|
@@ -91,14 +91,17 @@ def shorten_and_generate_qr(api_key, text, shorten, generate_qr):
|
|
91 |
bitly = ShortenUrl(api_key)
|
92 |
urls = re.findall(r'http[s]?://\S+', text)
|
93 |
results = []
|
|
|
94 |
|
95 |
markdown_links = text
|
96 |
all_json_responses = []
|
97 |
qr_svgs = []
|
98 |
qr_pngs = []
|
99 |
|
100 |
-
for url in urls:
|
101 |
shorten_url, shorten_response_json = None, None
|
|
|
|
|
102 |
try:
|
103 |
if shorten:
|
104 |
shorten_url, shorten_response_json = bitly.shorten(url)
|
@@ -117,7 +120,8 @@ def shorten_and_generate_qr(api_key, text, shorten, generate_qr):
|
|
117 |
|
118 |
if qr_png:
|
119 |
markdown_links = markdown_links.replace(
|
120 |
-
url, f'{qr_html} [{url_to_process}]({url}) '
|
|
|
121 |
)
|
122 |
qr_pngs.append(qr_png)
|
123 |
qr_svgs.append(qr_html)
|
@@ -128,7 +132,8 @@ def shorten_and_generate_qr(api_key, text, shorten, generate_qr):
|
|
128 |
qr_html, qr_png, qr_json = bitly.generate_qr_local(url)
|
129 |
if qr_png:
|
130 |
markdown_links = markdown_links.replace(
|
131 |
-
url, f'{qr_html} [{url}]({url}) '
|
|
|
132 |
)
|
133 |
qr_pngs.append(qr_png)
|
134 |
qr_svgs.append(qr_html)
|
@@ -146,7 +151,6 @@ def shorten_and_generate_qr(api_key, text, shorten, generate_qr):
|
|
146 |
results.append((str(e), url))
|
147 |
|
148 |
try:
|
149 |
-
ts, tmpd = Interface.get_tempdir()
|
150 |
json_output_path = os.path.join(tmpd, 'response.json')
|
151 |
with open(json_output_path, 'w') as json_file:
|
152 |
json.dump(all_json_responses, json_file)
|
|
|
79 |
buffered = BytesIO()
|
80 |
img.save(buffered, format="PNG")
|
81 |
qr_code_png = base64.b64encode(buffered.getvalue())
|
82 |
+
img_str = qr_code_png.decode('ascii')
|
83 |
|
84 |
+
qr_code_html = f'<nobr><img src="data:image/png;base64,{img_str}" alt="QR Code" height="40px"></nobr>'
|
85 |
json_str = {"url": url}
|
86 |
return qr_code_html, buffered.getvalue(), json_str
|
87 |
except Exception as e:
|
|
|
91 |
bitly = ShortenUrl(api_key)
|
92 |
urls = re.findall(r'http[s]?://\S+', text)
|
93 |
results = []
|
94 |
+
ts, tmpd = Interface.get_tempdir()
|
95 |
|
96 |
markdown_links = text
|
97 |
all_json_responses = []
|
98 |
qr_svgs = []
|
99 |
qr_pngs = []
|
100 |
|
101 |
+
for idx, url in enumerate(urls):
|
102 |
shorten_url, shorten_response_json = None, None
|
103 |
+
png_file_name = f'qr_code_{idx}.png'
|
104 |
+
png_output_path = os.path.join(tmpd, png_file_name)
|
105 |
try:
|
106 |
if shorten:
|
107 |
shorten_url, shorten_response_json = bitly.shorten(url)
|
|
|
120 |
|
121 |
if qr_png:
|
122 |
markdown_links = markdown_links.replace(
|
123 |
+
# url, f'{qr_html} [{url_to_process}]({url}) '
|
124 |
+
url, f'<img src="{png_file_name}" height="40"> [{url}]({url_to_process}) '
|
125 |
)
|
126 |
qr_pngs.append(qr_png)
|
127 |
qr_svgs.append(qr_html)
|
|
|
132 |
qr_html, qr_png, qr_json = bitly.generate_qr_local(url)
|
133 |
if qr_png:
|
134 |
markdown_links = markdown_links.replace(
|
135 |
+
# url, f'{qr_html} [{url}]({url}) '
|
136 |
+
url, f'<img src="{png_file_name}" height="40") [{url}]({url}) '
|
137 |
)
|
138 |
qr_pngs.append(qr_png)
|
139 |
qr_svgs.append(qr_html)
|
|
|
151 |
results.append((str(e), url))
|
152 |
|
153 |
try:
|
|
|
154 |
json_output_path = os.path.join(tmpd, 'response.json')
|
155 |
with open(json_output_path, 'w') as json_file:
|
156 |
json.dump(all_json_responses, json_file)
|