added purge for source (#6)

* added purge for source * pre-commit fix --------- Co-authored-by: Davit Shartava <[email protected]>
rossumai · Aug 16, 2024 · 5ac751e · 5ac751e
1 parent 3e748c2
commit 5ac751e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/project_rossum_deploy/commands/purge/purge.py b/project_rossum_deploy/commands/purge/purge.py
@@ -30,7 +30,9 @@
 )
 @click.argument(
     "destination",
-    type=click.Choice([settings.TARGET_DIRNAME, settings.UNUSED_SCHEMAS]),
+    type=click.Choice(
+        [settings.SOURCE_DIRNAME, settings.TARGET_DIRNAME, settings.UNUSED_SCHEMAS]
+    ),
 )
 @coro
 async def purge_project_wrapper(destination):