admin commited on
Commit
9d75477
·
1 Parent(s): 022aaa4
Files changed (1) hide show
  1. 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)