pearmut 0.2.5__tar.gz → 0.2.6__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {pearmut-0.2.5 → pearmut-0.2.6}/PKG-INFO +12 -7
- {pearmut-0.2.5 → pearmut-0.2.6}/README.md +11 -6
- {pearmut-0.2.5 → pearmut-0.2.6}/pearmut.egg-info/PKG-INFO +12 -7
- {pearmut-0.2.5 → pearmut-0.2.6}/pyproject.toml +1 -1
- {pearmut-0.2.5 → pearmut-0.2.6}/server/cli.py +119 -13
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/listwise.bundle.js +1 -1
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/pointwise.bundle.js +1 -1
- {pearmut-0.2.5 → pearmut-0.2.6}/LICENSE +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/pearmut.egg-info/SOURCES.txt +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/pearmut.egg-info/dependency_links.txt +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/pearmut.egg-info/entry_points.txt +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/pearmut.egg-info/requires.txt +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/pearmut.egg-info/top_level.txt +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/app.py +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/assignment.py +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/assets/favicon.svg +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/assets/style.css +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/dashboard.bundle.js +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/dashboard.html +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/index.html +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/listwise.html +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/static/pointwise.html +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/server/utils.py +0 -0
- {pearmut-0.2.5 → pearmut-0.2.6}/setup.cfg +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: pearmut
|
|
3
|
-
Version: 0.2.
|
|
3
|
+
Version: 0.2.6
|
|
4
4
|
Summary: A tool for evaluation of model outputs, primarily MT.
|
|
5
5
|
Author-email: Vilém Zouhar <vilem.zouhar@gmail.com>
|
|
6
6
|
License: apache-2.0
|
|
@@ -67,7 +67,6 @@ One of the simplest ones, where each user has a pre-defined list of tasks (`task
|
|
|
67
67
|
"protocol_score": true, # we want scores [0...100] for each segment
|
|
68
68
|
"protocol_error_spans": true, # we want error spans
|
|
69
69
|
"protocol_error_categories": false, # we do not want error span categories
|
|
70
|
-
"instructions": "Evaluate translation from en to cs_CZ", # message to show to users
|
|
71
70
|
},
|
|
72
71
|
"campaign_id": "wmt25_#_en-cs_CZ",
|
|
73
72
|
"data": [
|
|
@@ -76,6 +75,7 @@ One of the simplest ones, where each user has a pre-defined list of tasks (`task
|
|
|
76
75
|
[
|
|
77
76
|
# each evaluation item is a document
|
|
78
77
|
{
|
|
78
|
+
"instructions": "Evaluate translation from en to cs_CZ", # message to show to users above the first item
|
|
79
79
|
"src": "This will be the year that Guinness loses its cool. Cheers to that!",
|
|
80
80
|
"tgt": "Nevím přesně, kdy jsem to poprvé zaznamenal. Možná to bylo ve chvíli, ..."
|
|
81
81
|
},
|
|
@@ -288,22 +288,25 @@ Pearmut provides the following commands:
|
|
|
288
288
|
|
|
289
289
|
## Hosting Assets
|
|
290
290
|
|
|
291
|
-
|
|
292
|
-
When present, this directory is symlinked to the `static/` directory so its contents become accessible from the server.
|
|
291
|
+
To host local assets (e.g., audio files, images, videos), use the `assets` key in your campaign file:
|
|
293
292
|
|
|
294
293
|
```python
|
|
295
294
|
{
|
|
296
295
|
"campaign_id": "my_campaign",
|
|
297
296
|
"info": {
|
|
298
|
-
"assets":
|
|
297
|
+
"assets": {
|
|
298
|
+
"source": "videos", # path to directory containing assets
|
|
299
|
+
"destination": "assets/my_videos" # where to mount (must start with "assets/")
|
|
300
|
+
},
|
|
299
301
|
...
|
|
300
302
|
},
|
|
301
303
|
"data": [ ... ]
|
|
302
304
|
}
|
|
303
305
|
```
|
|
304
306
|
|
|
305
|
-
|
|
306
|
-
|
|
307
|
+
Files in `videos/` become accessible at `localhost:8001/assets/my_videos/`.
|
|
308
|
+
This creates a symlink, so the source directory must exist throughout the annotation period.
|
|
309
|
+
If another campaign already uses the destination path, the add command will fail.
|
|
307
310
|
|
|
308
311
|
## Development
|
|
309
312
|
|
|
@@ -333,6 +336,8 @@ The `pearmut run` also accepts `--port` (default 8001).
|
|
|
333
336
|
If you wish to create a new protocol (referenceable from `info->template`), simply create a new HTML and TS file in `web/src` and add a rule to `webpack.config.js` so that your template gets built.
|
|
334
337
|
A template can call the server for data etc (see [web/src/pointwise.ts](web/src/pointwise.ts) as an exmple).
|
|
335
338
|
|
|
339
|
+
To run the platform, you need to run this on some publicly-facing server, or run it locally and tunnel your port to a public IP/domain.
|
|
340
|
+
|
|
336
341
|
## Citation
|
|
337
342
|
|
|
338
343
|
If you use this work in your paper, please cite as:
|
|
@@ -47,7 +47,6 @@ One of the simplest ones, where each user has a pre-defined list of tasks (`task
|
|
|
47
47
|
"protocol_score": true, # we want scores [0...100] for each segment
|
|
48
48
|
"protocol_error_spans": true, # we want error spans
|
|
49
49
|
"protocol_error_categories": false, # we do not want error span categories
|
|
50
|
-
"instructions": "Evaluate translation from en to cs_CZ", # message to show to users
|
|
51
50
|
},
|
|
52
51
|
"campaign_id": "wmt25_#_en-cs_CZ",
|
|
53
52
|
"data": [
|
|
@@ -56,6 +55,7 @@ One of the simplest ones, where each user has a pre-defined list of tasks (`task
|
|
|
56
55
|
[
|
|
57
56
|
# each evaluation item is a document
|
|
58
57
|
{
|
|
58
|
+
"instructions": "Evaluate translation from en to cs_CZ", # message to show to users above the first item
|
|
59
59
|
"src": "This will be the year that Guinness loses its cool. Cheers to that!",
|
|
60
60
|
"tgt": "Nevím přesně, kdy jsem to poprvé zaznamenal. Možná to bylo ve chvíli, ..."
|
|
61
61
|
},
|
|
@@ -268,22 +268,25 @@ Pearmut provides the following commands:
|
|
|
268
268
|
|
|
269
269
|
## Hosting Assets
|
|
270
270
|
|
|
271
|
-
|
|
272
|
-
When present, this directory is symlinked to the `static/` directory so its contents become accessible from the server.
|
|
271
|
+
To host local assets (e.g., audio files, images, videos), use the `assets` key in your campaign file:
|
|
273
272
|
|
|
274
273
|
```python
|
|
275
274
|
{
|
|
276
275
|
"campaign_id": "my_campaign",
|
|
277
276
|
"info": {
|
|
278
|
-
"assets":
|
|
277
|
+
"assets": {
|
|
278
|
+
"source": "videos", # path to directory containing assets
|
|
279
|
+
"destination": "assets/my_videos" # where to mount (must start with "assets/")
|
|
280
|
+
},
|
|
279
281
|
...
|
|
280
282
|
},
|
|
281
283
|
"data": [ ... ]
|
|
282
284
|
}
|
|
283
285
|
```
|
|
284
286
|
|
|
285
|
-
|
|
286
|
-
|
|
287
|
+
Files in `videos/` become accessible at `localhost:8001/assets/my_videos/`.
|
|
288
|
+
This creates a symlink, so the source directory must exist throughout the annotation period.
|
|
289
|
+
If another campaign already uses the destination path, the add command will fail.
|
|
287
290
|
|
|
288
291
|
## Development
|
|
289
292
|
|
|
@@ -313,6 +316,8 @@ The `pearmut run` also accepts `--port` (default 8001).
|
|
|
313
316
|
If you wish to create a new protocol (referenceable from `info->template`), simply create a new HTML and TS file in `web/src` and add a rule to `webpack.config.js` so that your template gets built.
|
|
314
317
|
A template can call the server for data etc (see [web/src/pointwise.ts](web/src/pointwise.ts) as an exmple).
|
|
315
318
|
|
|
319
|
+
To run the platform, you need to run this on some publicly-facing server, or run it locally and tunnel your port to a public IP/domain.
|
|
320
|
+
|
|
316
321
|
## Citation
|
|
317
322
|
|
|
318
323
|
If you use this work in your paper, please cite as:
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: pearmut
|
|
3
|
-
Version: 0.2.
|
|
3
|
+
Version: 0.2.6
|
|
4
4
|
Summary: A tool for evaluation of model outputs, primarily MT.
|
|
5
5
|
Author-email: Vilém Zouhar <vilem.zouhar@gmail.com>
|
|
6
6
|
License: apache-2.0
|
|
@@ -67,7 +67,6 @@ One of the simplest ones, where each user has a pre-defined list of tasks (`task
|
|
|
67
67
|
"protocol_score": true, # we want scores [0...100] for each segment
|
|
68
68
|
"protocol_error_spans": true, # we want error spans
|
|
69
69
|
"protocol_error_categories": false, # we do not want error span categories
|
|
70
|
-
"instructions": "Evaluate translation from en to cs_CZ", # message to show to users
|
|
71
70
|
},
|
|
72
71
|
"campaign_id": "wmt25_#_en-cs_CZ",
|
|
73
72
|
"data": [
|
|
@@ -76,6 +75,7 @@ One of the simplest ones, where each user has a pre-defined list of tasks (`task
|
|
|
76
75
|
[
|
|
77
76
|
# each evaluation item is a document
|
|
78
77
|
{
|
|
78
|
+
"instructions": "Evaluate translation from en to cs_CZ", # message to show to users above the first item
|
|
79
79
|
"src": "This will be the year that Guinness loses its cool. Cheers to that!",
|
|
80
80
|
"tgt": "Nevím přesně, kdy jsem to poprvé zaznamenal. Možná to bylo ve chvíli, ..."
|
|
81
81
|
},
|
|
@@ -288,22 +288,25 @@ Pearmut provides the following commands:
|
|
|
288
288
|
|
|
289
289
|
## Hosting Assets
|
|
290
290
|
|
|
291
|
-
|
|
292
|
-
When present, this directory is symlinked to the `static/` directory so its contents become accessible from the server.
|
|
291
|
+
To host local assets (e.g., audio files, images, videos), use the `assets` key in your campaign file:
|
|
293
292
|
|
|
294
293
|
```python
|
|
295
294
|
{
|
|
296
295
|
"campaign_id": "my_campaign",
|
|
297
296
|
"info": {
|
|
298
|
-
"assets":
|
|
297
|
+
"assets": {
|
|
298
|
+
"source": "videos", # path to directory containing assets
|
|
299
|
+
"destination": "assets/my_videos" # where to mount (must start with "assets/")
|
|
300
|
+
},
|
|
299
301
|
...
|
|
300
302
|
},
|
|
301
303
|
"data": [ ... ]
|
|
302
304
|
}
|
|
303
305
|
```
|
|
304
306
|
|
|
305
|
-
|
|
306
|
-
|
|
307
|
+
Files in `videos/` become accessible at `localhost:8001/assets/my_videos/`.
|
|
308
|
+
This creates a symlink, so the source directory must exist throughout the annotation period.
|
|
309
|
+
If another campaign already uses the destination path, the add command will fail.
|
|
307
310
|
|
|
308
311
|
## Development
|
|
309
312
|
|
|
@@ -333,6 +336,8 @@ The `pearmut run` also accepts `--port` (default 8001).
|
|
|
333
336
|
If you wish to create a new protocol (referenceable from `info->template`), simply create a new HTML and TS file in `web/src` and add a rule to `webpack.config.js` so that your template gets built.
|
|
334
337
|
A template can call the server for data etc (see [web/src/pointwise.ts](web/src/pointwise.ts) as an exmple).
|
|
335
338
|
|
|
339
|
+
To run the platform, you need to run this on some publicly-facing server, or run it locally and tunnel your port to a public IP/domain.
|
|
340
|
+
|
|
336
341
|
## Citation
|
|
337
342
|
|
|
338
343
|
If you use this work in your paper, please cite as:
|
|
@@ -12,6 +12,9 @@ import psutil
|
|
|
12
12
|
|
|
13
13
|
from .utils import ROOT, load_progress_data, save_progress_data
|
|
14
14
|
|
|
15
|
+
# Static directory path (constant for consistency)
|
|
16
|
+
STATIC_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "static")
|
|
17
|
+
|
|
15
18
|
os.makedirs(f"{ROOT}/data/tasks", exist_ok=True)
|
|
16
19
|
load_progress_data(warn=None)
|
|
17
20
|
|
|
@@ -50,6 +53,40 @@ def _run(args_unknown):
|
|
|
50
53
|
)
|
|
51
54
|
|
|
52
55
|
|
|
56
|
+
def _validate_item_structure(items, template):
|
|
57
|
+
"""
|
|
58
|
+
Validate that items have the correct structure.
|
|
59
|
+
Items should be lists of dictionaries with 'src' and 'tgt' keys.
|
|
60
|
+
|
|
61
|
+
Args:
|
|
62
|
+
items: List of item dictionaries to validate
|
|
63
|
+
template: Template type ('pointwise' or 'listwise') for type validation
|
|
64
|
+
"""
|
|
65
|
+
if not isinstance(items, list):
|
|
66
|
+
raise ValueError("Items must be a list")
|
|
67
|
+
|
|
68
|
+
for item in items:
|
|
69
|
+
if not isinstance(item, dict):
|
|
70
|
+
raise ValueError("Each item must be a dictionary with 'src' and 'tgt' keys")
|
|
71
|
+
if 'src' not in item or 'tgt' not in item:
|
|
72
|
+
raise ValueError("Each item must contain 'src' and 'tgt' keys")
|
|
73
|
+
|
|
74
|
+
# Validate src is always a string
|
|
75
|
+
if not isinstance(item['src'], str):
|
|
76
|
+
raise ValueError("Item 'src' must be a string")
|
|
77
|
+
|
|
78
|
+
# Validate tgt type based on template
|
|
79
|
+
if template == 'listwise':
|
|
80
|
+
if not isinstance(item['tgt'], list):
|
|
81
|
+
raise ValueError("Item 'tgt' must be a list for listwise template")
|
|
82
|
+
# Check that all elements in tgt list are strings
|
|
83
|
+
if not all(isinstance(t, str) for t in item['tgt']):
|
|
84
|
+
raise ValueError("All elements in 'tgt' list must be strings for listwise template")
|
|
85
|
+
elif template == 'pointwise':
|
|
86
|
+
if not isinstance(item['tgt'], str):
|
|
87
|
+
raise ValueError("Item 'tgt' must be a string for pointwise template")
|
|
88
|
+
|
|
89
|
+
|
|
53
90
|
def _add_single_campaign(data_file, overwrite, server):
|
|
54
91
|
"""
|
|
55
92
|
Add a single campaign from a JSON data file.
|
|
@@ -80,6 +117,7 @@ def _add_single_campaign(data_file, overwrite, server):
|
|
|
80
117
|
raise ValueError("Campaign 'info' must contain 'template' field.")
|
|
81
118
|
|
|
82
119
|
assignment = campaign_data["info"]["assignment"]
|
|
120
|
+
template = campaign_data["info"]["template"]
|
|
83
121
|
# use random words for identifying users
|
|
84
122
|
rng = random.Random(campaign_data["campaign_id"])
|
|
85
123
|
rword = wonderwords.RandomWord(rng=rng)
|
|
@@ -96,6 +134,13 @@ def _add_single_campaign(data_file, overwrite, server):
|
|
|
96
134
|
if not all(isinstance(task, list) for task in tasks):
|
|
97
135
|
raise ValueError(
|
|
98
136
|
"Each task in task-based campaign 'data' must be a list of items.")
|
|
137
|
+
# Validate item structure for each task
|
|
138
|
+
for task_i, task in enumerate(tasks):
|
|
139
|
+
for doc_i, doc in enumerate(task):
|
|
140
|
+
try:
|
|
141
|
+
_validate_item_structure(doc, template)
|
|
142
|
+
except ValueError as e:
|
|
143
|
+
raise ValueError(f"Task {task_i}, document {doc_i}: {e}")
|
|
99
144
|
num_users = len(tasks)
|
|
100
145
|
elif assignment == "single-stream":
|
|
101
146
|
tasks = campaign_data["data"]
|
|
@@ -105,6 +150,12 @@ def _add_single_campaign(data_file, overwrite, server):
|
|
|
105
150
|
if not isinstance(campaign_data["data"], list):
|
|
106
151
|
raise ValueError(
|
|
107
152
|
"Single-stream campaign 'data' must be a list of items.")
|
|
153
|
+
# Validate item structure for single-stream
|
|
154
|
+
for doc_i, doc in enumerate(tasks):
|
|
155
|
+
try:
|
|
156
|
+
_validate_item_structure(doc, template)
|
|
157
|
+
except ValueError as e:
|
|
158
|
+
raise ValueError(f"Document {doc_i}: {e}")
|
|
108
159
|
if isinstance(users_spec, int):
|
|
109
160
|
num_users = users_spec
|
|
110
161
|
elif isinstance(users_spec, list):
|
|
@@ -199,30 +250,62 @@ def _add_single_campaign(data_file, overwrite, server):
|
|
|
199
250
|
|
|
200
251
|
# Handle assets symlink if specified
|
|
201
252
|
if "assets" in campaign_data["info"]:
|
|
202
|
-
|
|
253
|
+
assets_config = campaign_data["info"]["assets"]
|
|
254
|
+
|
|
255
|
+
# assets must be a dictionary with source and destination keys
|
|
256
|
+
if not isinstance(assets_config, dict):
|
|
257
|
+
raise ValueError("Assets must be a dictionary with 'source' and 'destination' keys.")
|
|
258
|
+
if "source" not in assets_config or "destination" not in assets_config:
|
|
259
|
+
raise ValueError("Assets config must contain 'source' and 'destination' keys.")
|
|
260
|
+
|
|
261
|
+
assets_source = assets_config["source"]
|
|
262
|
+
assets_destination = assets_config["destination"]
|
|
263
|
+
|
|
264
|
+
# Validate destination starts with 'assets/'
|
|
265
|
+
if not assets_destination.startswith("assets/"):
|
|
266
|
+
raise ValueError(f"Assets destination '{assets_destination}' must start with 'assets/'.")
|
|
203
267
|
|
|
204
268
|
# Resolve relative paths from the caller's current working directory
|
|
205
|
-
assets_real_path = os.path.abspath(
|
|
269
|
+
assets_real_path = os.path.abspath(assets_source)
|
|
206
270
|
|
|
207
271
|
if not os.path.isdir(assets_real_path):
|
|
208
|
-
raise ValueError(f"Assets path '{assets_real_path}' must be an existing directory.")
|
|
272
|
+
raise ValueError(f"Assets source path '{assets_real_path}' must be an existing directory.")
|
|
209
273
|
|
|
210
|
-
|
|
211
|
-
dir_name = assets_real_path.split(os.sep)[-1]
|
|
212
|
-
|
|
213
|
-
if not os.path.isdir(static_dir):
|
|
274
|
+
if not os.path.isdir(STATIC_DIR):
|
|
214
275
|
raise ValueError(
|
|
215
|
-
f"Static directory '{
|
|
276
|
+
f"Static directory '{STATIC_DIR}' does not exist. "
|
|
216
277
|
"Please build the frontend first."
|
|
217
278
|
)
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
279
|
+
|
|
280
|
+
# Symlink path is based on the destination, stripping the 'assets/' prefix
|
|
281
|
+
symlink_path = f"{STATIC_DIR}/{assets_destination}"
|
|
282
|
+
|
|
283
|
+
# Remove existing symlink if present and we are overriding the same campaign
|
|
284
|
+
if os.path.lexists(symlink_path):
|
|
285
|
+
# Check if any other campaign is using this destination
|
|
286
|
+
current_campaign_id = campaign_data['campaign_id']
|
|
287
|
+
tasks_dir = f"{ROOT}/data/tasks"
|
|
288
|
+
if os.path.exists(tasks_dir):
|
|
289
|
+
for task_file in os.listdir(tasks_dir):
|
|
290
|
+
if task_file.endswith('.json'):
|
|
291
|
+
other_campaign_id = task_file[:-5]
|
|
292
|
+
if other_campaign_id != current_campaign_id:
|
|
293
|
+
with open(f"{tasks_dir}/{task_file}", "r") as f:
|
|
294
|
+
other_campaign = json.load(f)
|
|
295
|
+
other_assets = other_campaign.get("info", {}).get("assets")
|
|
296
|
+
if other_assets and isinstance(other_assets, dict):
|
|
297
|
+
if other_assets.get("destination") == assets_destination:
|
|
298
|
+
raise ValueError(
|
|
299
|
+
f"Assets destination '{assets_destination}' is already used by campaign '{other_campaign_id}'."
|
|
300
|
+
)
|
|
301
|
+
# Only allow overwrite if it's the same campaign
|
|
222
302
|
if overwrite:
|
|
223
303
|
os.remove(symlink_path)
|
|
224
304
|
else:
|
|
225
|
-
raise ValueError(f"Assets
|
|
305
|
+
raise ValueError(f"Assets destination '{assets_destination}' is already taken.")
|
|
306
|
+
|
|
307
|
+
# Ensure the assets directory exists
|
|
308
|
+
os.makedirs(f"{STATIC_DIR}/assets", exist_ok=True)
|
|
226
309
|
|
|
227
310
|
os.symlink(assets_real_path, symlink_path, target_is_directory=True)
|
|
228
311
|
print(f"Assets symlinked: {symlink_path} -> {assets_real_path}")
|
|
@@ -299,6 +382,20 @@ def main():
|
|
|
299
382
|
elif args.command == 'purge':
|
|
300
383
|
import shutil
|
|
301
384
|
|
|
385
|
+
def _unlink_assets(campaign_id):
|
|
386
|
+
"""Unlink assets symlink for a campaign if it exists."""
|
|
387
|
+
task_file = f"{ROOT}/data/tasks/{campaign_id}.json"
|
|
388
|
+
if not os.path.exists(task_file):
|
|
389
|
+
return
|
|
390
|
+
with open(task_file, "r") as f:
|
|
391
|
+
campaign_data = json.load(f)
|
|
392
|
+
destination = campaign_data.get("info", {}).get("assets", {}).get("destination")
|
|
393
|
+
if destination:
|
|
394
|
+
symlink_path = f"{STATIC_DIR}/{destination}"
|
|
395
|
+
if os.path.islink(symlink_path):
|
|
396
|
+
os.remove(symlink_path)
|
|
397
|
+
print(f"Assets symlink removed: {symlink_path}")
|
|
398
|
+
|
|
302
399
|
# Parse optional campaign name
|
|
303
400
|
purge_args = argparse.ArgumentParser()
|
|
304
401
|
purge_args.add_argument(
|
|
@@ -314,6 +411,8 @@ def main():
|
|
|
314
411
|
f"Are you sure you want to purge campaign '{campaign_id}'? This action cannot be undone. [y/n] "
|
|
315
412
|
)
|
|
316
413
|
if confirm.lower() == 'y':
|
|
414
|
+
# Unlink assets before removing task file
|
|
415
|
+
_unlink_assets(campaign_id)
|
|
317
416
|
# Remove task file
|
|
318
417
|
task_file = f"{ROOT}/data/tasks/{campaign_id}.json"
|
|
319
418
|
if os.path.exists(task_file):
|
|
@@ -336,6 +435,13 @@ def main():
|
|
|
336
435
|
"Are you sure you want to purge all campaign data? This action cannot be undone. [y/n] "
|
|
337
436
|
)
|
|
338
437
|
if confirm.lower() == 'y':
|
|
438
|
+
# Unlink all assets first
|
|
439
|
+
tasks_dir = f"{ROOT}/data/tasks"
|
|
440
|
+
if os.path.exists(tasks_dir):
|
|
441
|
+
for task_file in os.listdir(tasks_dir):
|
|
442
|
+
if task_file.endswith('.json'):
|
|
443
|
+
campaign_id = task_file[:-5]
|
|
444
|
+
_unlink_assets(campaign_id)
|
|
339
445
|
shutil.rmtree(f"{ROOT}/data/tasks", ignore_errors=True)
|
|
340
446
|
shutil.rmtree(f"{ROOT}/data/outputs", ignore_errors=True)
|
|
341
447
|
if os.path.exists(f"{ROOT}/data/progress.json"):
|