Update imports and auto register tasks in worker.py

backend/app/app/source/ICLR.py

@@ -1,4 +1,4 @@
-from app.source import NIPS
+from app.source.NIPS import NIPS
 
 
 class ICLR(NIPS):

backend/app/app/source/ICML.py

@@ -1,4 +1,4 @@
-from app.source import NIPS
+from app.source.NIPS import NIPS
 
 
 class ICML(NIPS):

backend/app/app/source/__init__.py

@@ -1,7 +1,6 @@
 from app.source.Arxiv import Arxiv
-from app.source.base import PaperRequestsTask
 from app.source.ICLR import ICLR
 from app.source.ICML import ICML
 from app.source.NIPS import NIPS
 
-__all__ = ["PaperRequestsTask", "NIPS", "ICLR", "ICML", "Arxiv"]
+__all__ = ["NIPS", "ICLR", "ICML", "Arxiv"]

backend/app/app/worker.py

@@ -1,13 +1,16 @@
+import inspect
 from typing import Union
 
 from celery import Task
 from sqlmodel import Session
 
+from app import source
 from app.core.celery_app import celery_app
-from app.source import NIPS, Arxiv
 
-celery_app.register_task(NIPS())
-celery_app.register_task(Arxiv())
+members = inspect.getmembers(source, inspect.isclass)
+for name, _class in members:
+    # auto register task
+    celery_app.register_task(_class)
 
 
 def batch(iterable: Union[set[str], list[str]], n: int = 1):