xarical commited on
Commit
98ce236
·
verified ·
1 Parent(s): 67557fb

Upload utils.py

Browse files
Files changed (1) hide show
  1. utils.py +45 -0
utils.py ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from collections import OrderedDict
2
+ from threading import Thread
3
+
4
+ from flask import Flask
5
+
6
+ flask_app = Flask(__name__)
7
+
8
+
9
+ @flask_app.route("/", methods=["GET"])
10
+ def home() -> str:
11
+ """
12
+ Keepalive endpoint for hosting services.
13
+ """
14
+ return "I'm alive"
15
+
16
+
17
+ def keepalive_run() -> None:
18
+ """
19
+ Run keepalive Flask app in a daemon thread.
20
+ """
21
+ def flask_run():
22
+ flask_app.run(host="0.0.0.0", port="7860")
23
+ flask_thread = Thread(target=flask_run, daemon=True)
24
+ flask_thread.start()
25
+
26
+
27
+ class LRUCache:
28
+ """Least Recently Used (LRU) cache prototype implementation"""
29
+ def __init__(self, max_size):
30
+ self.max_size = max_size
31
+ self.cache = OrderedDict()
32
+
33
+ def __getitem__(self, key):
34
+ try:
35
+ val = self.cache[key]
36
+ self.cache.move_to_end(key)
37
+ return val
38
+ except KeyError:
39
+ return None
40
+
41
+ def __setitem__(self, key, value):
42
+ self.cache[key] = value
43
+ self.cache.move_to_end(key)
44
+ if len(self.cache) > self.max_size:
45
+ self.cache.popitem(last=False)