Jelajahi Sumber

use correct 'person' parameter when saving content

Daniel Sheffield 1 tahun lalu
induk
melakukan
157eff066f
2 mengubah file dengan 5 tambahan dan 5 penghapusan
  1. 2 2
      rest/pyapi.py
  2. 3 3
      rest/save.py

+ 2 - 2
rest/pyapi.py

@@ -60,7 +60,7 @@ def clip():
     if request.method == 'POST':
         content = validate_parameter(request, 'paste')
         if request.params.copy != 'true':
-            _b32 = save(content, LOCATION, root='rest/static/files')
+            _b32 = save(content, 'clip', LOCATION, root='rest/static/files')
             return redirect(f'/clip?hash={_b32}')
 
         response.content_type = 'text/html; charset=utf-8'
@@ -219,7 +219,7 @@ def goto():
 
     if request.method == 'POST':
         content = validate_parameter(request, 'url')
-        _b32 = save(content, LOCATION, root='rest/static/files')
+        _b32 = save(content, 'goto', LOCATION, root='rest/static/files')
 
         # validate but save content unmodified
         _ = validate_url(content.decode('utf-8'))

+ 3 - 3
rest/save.py

@@ -19,8 +19,8 @@ def save_qr(qr: bytes, _b32:str , directory: str) -> str:
         f.write(qr)
 
 
-def save(content: bytes, location: str, root='rest/static/files') -> str:
-    _bytes = blake(content, person='clip'.encode('utf-8'))
+def save(content: bytes, tool: str, location: str, root='rest/static/files') -> str:
+    _bytes = blake(content, person=tool.encode('utf-8'))
     _b32 = bytes_to_base32(_bytes)
     directory = f'{root}/{_b32}'
     try:
@@ -31,7 +31,7 @@ def save(content: bytes, location: str, root='rest/static/files') -> str:
     with open(fd, "wb") as f:
         f.write(content)
     
-    link = f'{location}/clip/{_b32}'
+    link = f'{location}/{tool}/{_b32}'
     svg = get_qr_code(content, fallback=link)
     save_qr(svg, _b32, directory)
     return _b32