10 Commits

Author SHA1 Message Date
ba68674e4e Add GitHub-specific README.md which links Gitea 2020-09-07 20:31:24 -05:00
99ff2c68a3 Change references of gitea.bbaovanc.com to git.bbaovanc.com 2020-09-07 02:07:36 -05:00
dd069bf395 Add newline after block comments at beginning 2020-09-06 15:24:20 -05:00
f21adfa04e Added FAQ and links to sections in README.md 2020-09-06 12:43:44 -05:00
3d5c55498f Clean up and add LICENSE section to README.md 2020-09-06 11:26:13 -05:00
5e2be10434 Change references of GitHub to gitea.bbaovanc.com 2020-09-06 11:09:06 -05:00
7c1f449bce Add "verify" field to request to not save image
This makes it easy for the user to debug authentication.
2020-09-05 18:55:56 -05:00
0dbcc0e380 Change file extension check to be case-insensitive 2020-09-05 16:21:50 -05:00
b8b5a2518c Change abort() calls to JSON responses
This makes the responses more consistent. Now, all responses are JSON.
2020-09-05 15:43:36 -05:00
805e545b39 Deduplicate code in keyctl.py and add comments 2020-09-04 19:44:46 -05:00
6 changed files with 87 additions and 40 deletions

7
.github/README.md vendored Normal file
View File

@ -0,0 +1,7 @@
# imgupload
## Moving from GitHub to Gitea
**TL;DR: Please go to my Gitea instance instead of GitHub for anything related to imgupload. [https://git.bbaovanc.com/bbaovanc/imgupload](https://git.bbaovanc.com/bbaovanc/imgupload)**
This repository might not exist on GitHub in the future! Releases will not be released here in the future. Instead, they will be released on the repository on my Gitea instance, which you can find [here](https://git.bbaovanc.com/bbaovanc/imgupload). Issues and pull requests should also be created on Gitea. For now, commits will still be pushed to this repository, but that may change in the future.

View File

@ -1,26 +1,63 @@
# imgupload # imgupload
![CodeFactor Grade](https://img.shields.io/codefactor/grade/github/BBaoVanC/imgupload/master?color=purple) ![GitHub repo size](https://img.shields.io/github/repo-size/bbaovanc/imgupload?color=purple) ![GitHub All Releases](https://img.shields.io/github/downloads/bbaovanc/imgupload/total?color=purple) ![GitHub issues](https://img.shields.io/github/issues/bbaovanc/imgupload?color=purple) ![GitHub closed issues](https://img.shields.io/github/issues-closed/bbaovanc/imgupload?color=purple) ![GitHub](https://img.shields.io/github/license/bbaovanc/imgupload?color=purple)
### What is imgupload? <!---![CodeFactor Grade](https://img.shields.io/codefactor/grade/github/BBaoVanC/imgupload/master?color=purple)
![GitHub repo size](https://img.shields.io/github/repo-size/bbaovanc/imgupload?color=purple)
![GitHub All Releases](https://img.shields.io/github/downloads/bbaovanc/imgupload/total?color=purple)
![GitHub issues](https://img.shields.io/github/issues/bbaovanc/imgupload?color=purple)
![GitHub closed issues](https://img.shields.io/github/issues-closed/bbaovanc/imgupload?color=purple)
![GitHub](https://img.shields.io/github/license/bbaovanc/imgupload?color=purple)-->
## What is imgupload?
imgupload is a Flask + uWSGI application to serve as an all-purpose image/file uploader over POST requests. imgupload is a Flask + uWSGI application to serve as an all-purpose image/file uploader over POST requests.
### Installation ---
1. Clone the repository: `git clone https://github.com/BBaoVanC/imgupload.git`
## FAQ
**Where can I send bug reports and feature requests?**
You can create an issue [here](https://git.bbaovanc.com/bbaovanc/imgupload/issues).
**How do I use this program?**
See [Installation](#installation)
**I want to make a pull request. Where should I do that?**
First, fork [this repository](https://git.bbaovanc.com/bbaovanc/imgupload). If you don't have an account on my Gitea site yet, you can either create one, or sign in using your GitHub account. Commit your changes to your fork, and then create a pull request.
---
## Installation
1. Clone the repository: `git clone https://git.bbaovanc.com/bbaovanc/imgupload.git`
2. Enter the imgupload directory: `cd imgupload` 2. Enter the imgupload directory: `cd imgupload`
3. Create a virtualenv: `python3 -m venv env` 3. Create a virtualenv: `python3 -m venv env`
4. Enter the virtualenv: `source env/bin/activate` 4. Enter the virtualenv: `source env/bin/activate`
5. Install dependencies: `python3 -m pip install -r requirements.txt` 5. Install dependencies: `python3 -m pip install -r requirements.txt`
6. Run the Flask app 6. [Run the Flask app](#running-the-flask-app)
---
## Running the Flask app ## Running the Flask app
### Using uWSGI ### Using uWSGI
[https://uwsgi-docs.readthedocs.io/en/latest/Configuration.html](https://uwsgi-docs.readthedocs.io/en/latest/Configuration.html) [https://uwsgi-docs.readthedocs.io/en/latest/Configuration.html](https://uwsgi-docs.readthedocs.io/en/latest/Configuration.html)
Instructions specific to imgupload are coming soon Instructions specific to imgupload are coming soon
### Using Flask development server ### Using Flask development server
```shell ```shell
$ source env/bin/activate # if you haven't already entered the virtualenv $ source env/bin/activate # if you haven't already entered the virtualenv
$ export FLASK_APP=imgupload.py $ export FLASK_APP=imgupload.py
$ flask run $ flask run
``` ```
---
## License
_imgupload_ is licensed under the GPLv3 license. For more information, please refer to [`LICENSE`](https://git.bbaovanc.com/bbaovanc/imgupload/src/branch/master/LICENSE)

View File

@ -4,6 +4,7 @@ configtest.py
Tests the validity of your configuration in settings.py. Tests the validity of your configuration in settings.py.
""" """
import os import os
import settings as settings import settings as settings

View File

@ -4,6 +4,7 @@ functions.py
Functions used by imgupload which can be easily customized. Functions used by imgupload which can be easily customized.
""" """
import string import string
import random import random

View File

@ -4,7 +4,8 @@ imgupload.py
Flask application for processing images uploaded through POST requests. Flask application for processing images uploaded through POST requests.
""" """
from flask import Flask, request, jsonify, abort, Response
from flask import Flask, request, jsonify, Response
from flask_api import status from flask_api import status
from pathlib import Path from pathlib import Path
import os import os
@ -17,7 +18,7 @@ app = Flask(__name__) # app is the app
def allowed_extension(testext): def allowed_extension(testext):
if testext in settings.ALLOWED_EXTENSIONS: if testext.lower() in settings.ALLOWED_EXTENSIONS:
return True return True
else: else:
return False return False
@ -49,6 +50,11 @@ def upload():
if request.form["uploadKey"] in validkeys: # check if uploadKey is valid if request.form["uploadKey"] in validkeys: # check if uploadKey is valid
print("Key is valid!") print("Key is valid!")
if "verify" in request.form.keys():
if request.form["verify"] == "true":
print("Request is asking if key is valid (it is)")
return jsonify({'status': 'key_valid'})
if "imageUpload" in request.files: # check if image to upload was provided if "imageUpload" in request.files: # check if image to upload was provided
f = request.files["imageUpload"] # f is the image to upload f = request.files["imageUpload"] # f is the image to upload
else: else:
@ -81,21 +87,16 @@ def upload():
else: # if the extension was invalid else: # if the extension was invalid
print("Uploaded extension is invalid!") print("Uploaded extension is invalid!")
abort(415) return jsonify({'status': 'error', 'error': 'INVALID_EXTENSION'}), status.HTTP_415_UNSUPPORTED_MEDIA_TYPE
else: # if the key was not valid else: # if the key was not valid
print("Key is invalid!") print("Key is invalid!")
print("Request key: {0}".format(request.form["uploadKey"])) print("Request key: {0}".format(request.form["uploadKey"]))
abort(401) return jsonify({'status': 'error', 'error': 'UNAUTHORIZED'}), status.HTTP_401_UNAUTHORIZED
else: # if uploadKey was not found in request body else: # if uploadKey was not found in request body
print("No uploadKey found in request!") print("No uploadKey found in request!")
abort(401) return jsonify({'status': 'error', 'error': 'UNAUTHORIZED'}), status.HTTP_401_UNAUTHORIZED
else: # if the request method wasn't post
print("Request method was not POST!")
abort(405)
if __name__ == "__main__": if __name__ == "__main__":
print("Run with `flask` or a WSGI server!") print("Run with `flask` or a WSGI server!")

View File

@ -12,34 +12,39 @@ import secrets
import string import string
def read_keyfile():
with open("uploadkeys", "r") as keyfile: # open uploadkeys
keys = keyfile.readlines() # read all the keys
logging.debug("Read uploadkeys")
keys = [x.strip("\n") for x in keys] # strip newlines from keys
logging.debug("Stripped newlines from keys")
return keys
def genkey(length): def genkey(length):
key = ''.join(secrets.choice(string.ascii_letters + string.digits) for x in range(length)) key = ''.join(secrets.choice(string.ascii_letters + string.digits) for x in range(length))
return key return key
def savekey(key): def savekey(key):
if not Path("uploadkeys").is_file(): if not Path("uploadkeys").is_file(): # if uploadkeys doesn't exist, log an info message
logging.info("uploadkeys file doesn't exist, it will be created.") logging.info("uploadkeys file doesn't exist, it will be created.")
with open("uploadkeys", "a+") as keyfile: with open("uploadkeys", "a+") as keyfile:
keyfile.write(str(key) + "\n") keyfile.write(str(key) + "\n") # add the key
logging.debug("Saved a key to uploadkeys: {0}".format(key)) logging.debug("Saved a key to uploadkeys: {0}".format(key))
def rmkey(delkey): def rmkey(delkey):
removedkey = False removedkey = False
with open("uploadkeys", "r") as keyfile: allkeys = read_keyfile()
allkeys = keyfile.readlines() if delkey in allkeys: # if the key to remove exists
logging.debug("Loaded all upload keys") allkeys.remove(delkey) # remove the first instance of the key
allkeys = [x.strip("\n") for x in allkeys]
logging.debug("Stripped keys")
if delkey in allkeys:
allkeys.remove(delkey)
removedkey = True removedkey = True
logging.debug("Removed one instance of the key") logging.debug("Removed one instance of the key")
with open("uploadkeys", "w") as keyfile: with open("uploadkeys", "w") as keyfile:
for k in allkeys: for k in allkeys:
keyfile.write(k + "\n") keyfile.write(k + "\n") # write the remaining keys
if removedkey: if removedkey:
return True return True
@ -48,11 +53,8 @@ def rmkey(delkey):
def find_duplicates(): def find_duplicates():
with open("uploadkeys", "r") as keyfile: allkeys = read_keyfile()
allkeys = keyfile.readlines()
logging.debug("Read all keys")
allkeys = [x.strip("\n") for x in allkeys]
logging.debug("Stripped newlines")
seen = set() seen = set()
ukeys = [] ukeys = []
dupkeys = [] dupkeys = []
@ -66,11 +68,7 @@ def find_duplicates():
def get_keys(): def get_keys():
with open("uploadkeys", "r") as keyfile: # load valid keys validkeys = read_keyfile()
validkeys = keyfile.readlines()
logging.debug("Read uploadkeys")
validkeys = [x.strip("\n") for x in validkeys]
logging.debug("Stripped newlines from validkeys")
while "" in validkeys: while "" in validkeys:
validkeys.remove("") validkeys.remove("")
logging.debug("Removed blank keys") logging.debug("Removed blank keys")
@ -85,7 +83,7 @@ def cmd_list(args):
for i in range(len(validkeys)): for i in range(len(validkeys)):
showkey = validkeys[i][:6] showkey = validkeys[i][:6]
if len(validkeys[i]) > 6: if len(validkeys[i]) > 6:
showkey += "..." showkey += "..." # add ellipses since the key was shortened in list
print(" [{0}] {1}".format(i+1, showkey)) print(" [{0}] {1}".format(i+1, showkey))
@ -121,7 +119,9 @@ def cmd_remove(args):
logging.info("No key was removed.") logging.info("No key was removed.")
def cmd_dedupe(args): def cmd_dedupe(args):
for d in find_duplicates(): dupes = find_duplicates()
if len(dupes) > 0:
for d in dupes:
r = rmkey(d) r = rmkey(d)
logging.debug(r) logging.debug(r)
logging.info("Removed duplicate key: {0}".format(d)) logging.info("Removed duplicate key: {0}".format(d))