sheppy/tmnf-replay-server
1
0
Fork 0
mirror of https://github.com/FAUSheppy/tmnf-replay-server.git synced 2026-04-26 22:12:28 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: allow setting endpoint s3

Browse Source
This commit is contained in:
Johnathen Sheppard
2026-03-12 18:25:40 +01:00
parent 417539f729
commit c7ac901dc6
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
server.py
View File

@@ -21,6 +21,8 @@ app = flask.Flask("TM Friends Replay Server")
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("DB_URL") or "sqlite:///sqlite.db" app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("DB_URL") or "sqlite:///sqlite.db"
app.config["AUTH_HEADER"] = os.environ.get("AUTH_HEADER") or "X-Forwarded-Preferred-Username" app.config["AUTH_HEADER"] = os.environ.get("AUTH_HEADER") or "X-Forwarded-Preferred-Username"
S3_ENDPOINT_URL = os.getenv("S3_ENDPOINT_URL")
db = SQLAlchemy(app) db = SQLAlchemy(app)
SEASON_ORDERING = ["Winter", "Spring", "Summer", "Fall"] SEASON_ORDERING = ["Winter", "Spring", "Summer", "Fall"]
@@ -499,9 +501,17 @@ def s3_enabled():
S3_BUCKET S3_BUCKET
]) ])
def get_s3_client():
kwargs = {}
if S3_ENDPOINT_URL:
kwargs["endpoint_url"] = S3_ENDPOINT_URL
return boto3.client("s3", **kwargs)
def upload_to_s3(local_path, replay): def upload_to_s3(local_path, replay):
s3 = boto3.client("s3") s3 = get_s3_client()
key = f"{replay.uploader}/{replay.filehash}/{os.path.basename(local_path)}" key = f"{replay.uploader}/{replay.filehash}/{os.path.basename(local_path)}"
s3.upload_file(local_path, S3_BUCKET, key) s3.upload_file(local_path, S3_BUCKET, key)
return key return key