app.py

# from datetime import datetime
import os

from flask import Flask, request, jsonify
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ['DATABASE_URL']
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)
        

class TinyWebDB(db.Model):
    __tablename__ = 'tinywebdb'
    tag = db.Column(db.String, primary_key=True, nullable=False)
    value = db.Column(db.String, nullable=False)
    # The 'date' column is needed for deleting older entries, so not really required
    # date = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)


db.create_all()
db.session.commit()

@app.route('/')
def main_method():
    return 'AsteroidDB works perfectly!'


# -------------------------
#  Store Value
#  - Store a value by using tag.
# -------------------------
@app.route('/store', methods=['POST'])
def store_a_value():
    tag = request.form['tag']
    value = request.form['value']
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if tag:
        if tag == 'dbpass':
            return return_error('Not possible to do any action to password record!')
        else:
            # --------------------
            if getpassword:
                password = request.form['pass']
                if password != getpassword.value:
                    return return_error('Wrong password!')
            # --------------------
            existing_tag = TinyWebDB.query.filter_by(tag=tag).first()
            if existing_tag:
                existing_tag.value = value
                db.session.commit()
            else:
                data = TinyWebDB(tag=tag, value=value)
                db.session.add(data)
                db.session.commit()
        return jsonify(action="STORED", tag=tag, value=value)
    return return_error('Tag is not specified!')


# -------------------------
#  Get Value
#  - Get the value from tag.
# -------------------------
@app.route('/get', methods=['POST'])
def get_value():
    tag = request.form['tag']
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if tag:
        if tag == 'dbpass':
            return return_error('Not possible to do any action to password record!')
        else:
            # --------------------
            if getpassword:
                password = request.form['pass']
                if password != getpassword.value:
                    return return_error('Wrong password!')
            # --------------------
            value = TinyWebDB.query.filter_by(tag=tag).first().value
            return jsonify(action="GOT", tag=tag, value=value)
    return return_error('Not found the tag!')


# -------------------------
#  Get All Data
#  - Return everything from database. This method doesn't include password record tag and value.
# -------------------------
@app.route('/auth/data', methods=['POST'])
def get_data():
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if getpassword:
        # --------------------
        password = request.form['pass']
        if password != getpassword.value:
            return return_error('Wrong password!')
        # --------------------
        tags = TinyWebDB.query.all()
        datalist = []
        for tg in tags:
            if tg.tag != 'dbpass':
                datalist.append([tg.tag, tg.value])
        return jsonify(action="DATA", data=datalist)
    return return_error('You need to set a password first to use this feature!')


# -------------------------
#  Get All Tags
#  - Return all tags from database. This method doesn't include password record tag.
# -------------------------
@app.route('/getall', methods=['POST'])
def get_all():
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if getpassword:
        # --------------------
        password = request.form['pass']
        if password != getpassword.value:
            return return_error('Wrong password!')
        # --------------------
    tags = TinyWebDB.query.all()
    taglist = []
    for tg in tags:
    	if tg.tag != 'dbpass': 
            taglist.append(tg.tag)
    return jsonify(action="TAGS", tag=taglist)


# -------------------------
#  Delete Record
#  - Delete a record from tag.
# -------------------------
@app.route('/delete', methods=['POST'])
def delete_entry():
    tag = request.form['tag']
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if tag:
        if tag == 'dbpass':
            return return_error('Not possible to do any action to password record!')
        else:
            # --------------------
            if getpassword:
                password = request.form['pass']
                if password != getpassword.value:
                    return return_error('Wrong password!')
            # --------------------
            deleted = TinyWebDB.query.filter_by(tag=tag).first()
            db.session.delete(deleted)
            db.session.commit()
            return jsonify(action="DELETED", tag=tag)
    return return_error('Not found the tag!')


# -------------------------
#  Format Database
#  - Deletes every record from database, and remove password protection.
# -------------------------
@app.route('/format', methods=['POST'])
def delete_all():
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if getpassword:
        # --------------------
        password = request.form['pass']
        if password != getpassword.value:
            return return_error('Wrong password!')
        # --------------------
    try:
        count = db.session.query(TinyWebDB).delete()
        db.session.commit()
        return jsonify(action="FORMATTED", count=count)
    except:
        db.session.rollback()
        return return_error('Something went wrong while performing this action.')
    


# -------------------------
#  Set/Change Password
#  - If you set a password, you need to type a password when you modify the database.
#  - The password will be saved in the same table along with other data called "dbpass".
#  - If you forgot the password, there is no way to recover it.
# -------------------------
@app.route('/auth/password', methods=['POST'])
def set_key():
    newpassword = request.form['newpass']
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if newpassword:
        if getpassword:
            oldpassword = request.form['oldpass']
            if getpassword.value == oldpassword:
                getpassword.value = newpassword
                db.session.commit()
                return jsonify(action="CHANGED PASSWORD", oldpass=oldpassword, newpass=newpassword)
            else:
                return return_error('Wrong old password!')
        else:
            data = TinyWebDB(tag='dbpass', value=newpassword)
            db.session.add(data)
            db.session.commit()
            return jsonify(action="SET PASSWORD", newpass=newpassword)
    return return_error('No new password is specified!')


# -------------------------
#  Remove Password
#  - If you type your current password, requests won't require pass parameter anymore. And your password will be deleted.
# -------------------------
@app.route('/auth/unlock', methods=['POST'])
def remove_key():
    password = request.form['pass']
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if getpassword:
        if getpassword.value == password:
            deleted = TinyWebDB.query.filter_by(tag='dbpass').first()
            db.session.delete(deleted)
            db.session.commit()
            return jsonify(action="DELETED PASSWORD", password=password)
        else:
        	return return_error('Wrong password!')
    return return_error('You need to set a password first to use this feature!')


# -------------------------
#  Is Password True?
#  - Useful for applications. Returns 'true' if password is correct. Otherwise; 'false'.
# -------------------------
@app.route('/istrue', methods=['POST'])
def is_true():
    password = request.form['pass']
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if getpassword:
        # --------------------
        if password != getpassword.value:
            return jsonify(action="IS CORRECT",result=False)
        # --------------------
    return jsonify(action="IS CORRECT",result=True)
        

# -------------------------
#  Count All Records
#  - Returns a number that tells you how many records there are in database. 
# -------------------------
@app.route('/count')
def count_all():
    tags = TinyWebDB.query.all()
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    resl = len(tags)
    if getpassword:
    	resl = resl - 1
    return jsonify(action="COUNT", count=resl)


# -------------------------
#  Version
#  - Shows the tagged version of AsteroidDB instance. 
# -------------------------
@app.route('/version')
def version():
    return jsonify(action="VERSION", result="1.1")

# -------------------------
#  Is Locked?
#  - Gives information about current lock status.
# -------------------------
@app.route('/islocked')
def is_locked():
    getpassword = TinyWebDB.query.filter_by(tag='dbpass').first()
    if getpassword:
        return jsonify(action="IS LOCKED",result=True)
    else:
        return jsonify(action="IS LOCKED",result=False)

@app.errorhandler(405)
def method_not_allowed(e):
    return jsonify(action="ERROR",result="Method is not allowed!"), 405

@app.errorhandler(404)
def not_found(e):
    return jsonify(action="ERROR",result="The requested URL was not found on the AsteroidDB instance!"), 404

@app.errorhandler(500)
def internal_error(error):
    db.session.rollback()
    return jsonify(action="ERROR",result="Internal server error!"), 500

# Returns error.
def return_error(message):
    response = jsonify(action="ERROR",result=message)
    response.status_code = 400
    return response
        

if __name__ == '__main__':
    app.run()