Skip to content

Commit

Permalink
add firebase
Browse files Browse the repository at this point in the history
  • Loading branch information
amin-codes committed Aug 10, 2021
1 parent b903236 commit 1d60bb4
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 5 deletions.
18 changes: 14 additions & 4 deletions app.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,17 @@
from sqlalchemy.dialects import registry
registry.register("cockroachdb", "cockroachdb.sqlalchemy.dialect",
"CockroachDBDialect")
import firebase_admin
from firebase_admin import credentials, db
from history_actions import HistoryActions

app = Flask(__name__)
app.secret_key = "asdfasfdasfdsafasddfsadfasdfsadfdas"
app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0
LIMIT = 10
database = HistoryActions(os.environ['DATABASE_URL'])
cred = credentials.Certificate(json.loads(os.environ['FIREBASE_KEY']))
firebase_admin.initialize_app(cred)

userHistory = defaultdict(list)

Expand All @@ -24,15 +28,21 @@ def history():
ip = request.environ['REMOTE_ADDR']
print("ip====",ip)
if request.method == 'GET':
res = userHistory["ip"] if "ip" in userHistory else []
res = db.reference('/ip').get() if "ip" in db.reference("/").get(shallow=True) else []
return jsonify({'ip_address': res}), 200
elif request.method == 'POST':
title = request.get_json().get('title', '')
code = request.get_json().get('code', '')
#session["ip"].append({'title': title, 'code': code})
userHistory["ip"].append({'title': title, 'code': code})
while len(userHistory["ip"]) > LIMIT:
userHistory["ip"].pop(0)
#userHistory["ip"].append({'title': title, 'code': code})
db.reference('/ip').update({'title': title, 'code': code})
sz = len(db.reference("/ip").get())
if sz > LIMIT:
diff = sz - LIMIT
keys = db.reference("/ip").get(shallow=True)
for v in keys:
if diff == 0: break
db.reference("/ip/" + v).delete()
return jsonify(success=True), 200
if __name__ == '__main__':
app.run(debug=True) #debug=True so that caching doesn't occur
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,5 @@ Flask-Script
Flask-SQLAlchemy
SQLAlchemy
sqlalchemy-cockroachdb
psycopg2
psycopg2
firebase-admin

0 comments on commit 1d60bb4

Please sign in to comment.