garvitcpp commited on
Commit
8b7704f
·
verified ·
1 Parent(s): 7a37c72

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +24 -24
app/main.py CHANGED
@@ -1,25 +1,25 @@
1
- from fastapi import FastAPI
2
- from apscheduler.schedulers.asyncio import AsyncIOScheduler
3
- from .routes import file_routes
4
- from . import create_app
5
- from .services.storage_service import StorageService
6
-
7
- app = create_app()
8
- storage_service = StorageService()
9
-
10
- # Register routes
11
- app.include_router(file_routes.router, prefix="/api")
12
-
13
- # Set up scheduler for cleanup
14
- scheduler = AsyncIOScheduler()
15
- scheduler.add_job(storage_service.cleanup_expired_files, 'interval', minutes=1) # Run every 5 minutes
16
- scheduler.start()
17
-
18
- @app.get("/")
19
- async def root():
20
- return {"message": "File Sharing API with auto-cleanup"}
21
-
22
- # Handle shutdown
23
- @app.on_event("shutdown")
24
- async def shutdown_event():
25
  scheduler.shutdown()
 
1
+ from fastapi import FastAPI
2
+ from apscheduler.schedulers.asyncio import AsyncIOScheduler
3
+ from .routes import file_routes
4
+ from . import create_app
5
+ from .services.storage_service import StorageService
6
+
7
+ app = create_app()
8
+ storage_service = StorageService()
9
+
10
+ # Register routes
11
+ app.include_router(file_routes.router, prefix="/api")
12
+
13
+ # Set up scheduler for cleanup
14
+ scheduler = AsyncIOScheduler()
15
+ scheduler.add_job(storage_service.cleanup_expired_files, 'interval', minutes=1) # Run every 5 minutes
16
+ scheduler.start()
17
+
18
+ @app.get("/")
19
+ async def root():
20
+ return {"message": "File Sharing API with auto-cleanup"}
21
+
22
+ # Handle shutdown
23
+ @app.on_event("shutdown")
24
+ async def shutdown_event():
25
  scheduler.shutdown()