Mbonea commited on
Commit
254f37b
·
1 Parent(s): 94f5ffa

added names

Browse files
Files changed (1) hide show
  1. App/Worker.py +11 -9
App/Worker.py CHANGED
@@ -20,10 +20,12 @@ celery.conf.update(
20
 
21
  @worker_process_init.connect
22
  def worker_process_init_handler(**kwargs):
23
- bot.start()
 
 
24
 
25
 
26
- @celery.task
27
  def create_json_file(assets: List[Assets], asset_dir: str):
28
  for asset in assets:
29
  filename = f"{asset.type.capitalize()}Sequences.json"
@@ -53,7 +55,7 @@ def download_with_wget(link, download_dir, filename):
53
  subprocess.run(["aria2c", link, "-d", download_dir, "-o", filename])
54
 
55
 
56
- @celery.task
57
  def copy_remotion_app(src: str, dest: str):
58
  shutil.copytree(src, dest)
59
 
@@ -62,7 +64,7 @@ def copy_remotion_app(src: str, dest: str):
62
  # create_symlink(source_dir, target_dir=dest, symlink_name="node_module")
63
 
64
 
65
- @celery.task
66
  def unsilence(directory: str):
67
  output_dir = os.path.join(directory, "out/video.mp4")
68
  shortered_dir = os.path.join(directory, "out/temp.mp4")
@@ -71,13 +73,13 @@ def unsilence(directory: str):
71
  os.rename(shortered_dir, output_dir)
72
 
73
 
74
- @celery.task
75
  def install_dependencies(directory: str):
76
  os.chdir(directory)
77
  os.system("npm install")
78
 
79
 
80
- @celery.task
81
  def download_assets(links: List[LinkInfo], temp_dir: str):
82
  public_dir = f"{temp_dir}/public"
83
  for link in links:
@@ -86,14 +88,14 @@ def download_assets(links: List[LinkInfo], temp_dir: str):
86
  download_with_wget(file_link, public_dir, file_name)
87
 
88
 
89
- @celery.task
90
  def render_video(directory: str, output_directory: str):
91
  os.chdir(directory)
92
  os.system(f"npm run build --output {output_directory}")
93
  print("complete")
94
 
95
 
96
- @celery.task
97
  def cleanup_temp_directory(
98
  temp_dir: str, output_dir: str, chat_id: int = -1002069945904
99
  ):
@@ -107,7 +109,7 @@ def cleanup_temp_directory(
107
  shutil.rmtree(temp_dir, ignore_errors=True)
108
 
109
 
110
- @celery.task
111
  def celery_task(video_task: EditorRequest):
112
  remotion_app_dir = os.path.join("/srv", "Remotion-app")
113
  project_id = str(uuid.uuid4())
 
20
 
21
  @worker_process_init.connect
22
  def worker_process_init_handler(**kwargs):
23
+ name = kwargs.get("name")
24
+ if name == "send":
25
+ bot.start()
26
 
27
 
28
+ @celery.task(name="CreateFile")
29
  def create_json_file(assets: List[Assets], asset_dir: str):
30
  for asset in assets:
31
  filename = f"{asset.type.capitalize()}Sequences.json"
 
55
  subprocess.run(["aria2c", link, "-d", download_dir, "-o", filename])
56
 
57
 
58
+ @celery.task(name="CopyRemotion")
59
  def copy_remotion_app(src: str, dest: str):
60
  shutil.copytree(src, dest)
61
 
 
64
  # create_symlink(source_dir, target_dir=dest, symlink_name="node_module")
65
 
66
 
67
+ @celery.task(name="Unsilence")
68
  def unsilence(directory: str):
69
  output_dir = os.path.join(directory, "out/video.mp4")
70
  shortered_dir = os.path.join(directory, "out/temp.mp4")
 
73
  os.rename(shortered_dir, output_dir)
74
 
75
 
76
+ @celery.task(name="InstallDependency")
77
  def install_dependencies(directory: str):
78
  os.chdir(directory)
79
  os.system("npm install")
80
 
81
 
82
+ @celery.task(name="DownloadAssets")
83
  def download_assets(links: List[LinkInfo], temp_dir: str):
84
  public_dir = f"{temp_dir}/public"
85
  for link in links:
 
88
  download_with_wget(file_link, public_dir, file_name)
89
 
90
 
91
+ @celery.task(name="RenderFile")
92
  def render_video(directory: str, output_directory: str):
93
  os.chdir(directory)
94
  os.system(f"npm run build --output {output_directory}")
95
  print("complete")
96
 
97
 
98
+ @celery.task(name="send")
99
  def cleanup_temp_directory(
100
  temp_dir: str, output_dir: str, chat_id: int = -1002069945904
101
  ):
 
109
  shutil.rmtree(temp_dir, ignore_errors=True)
110
 
111
 
112
+ @celery.task(name="All")
113
  def celery_task(video_task: EditorRequest):
114
  remotion_app_dir = os.path.join("/srv", "Remotion-app")
115
  project_id = str(uuid.uuid4())