nsfw mark api
This commit is contained in:
2
aiws.py
2
aiws.py
@@ -134,7 +134,7 @@ async def start_scheduler(service: GenerationService):
|
|||||||
try:
|
try:
|
||||||
logger.info("Running scheduler for stacked generation killing")
|
logger.info("Running scheduler for stacked generation killing")
|
||||||
await service.cleanup_stale_generations()
|
await service.cleanup_stale_generations()
|
||||||
await service.cleanup_old_data(days=5)
|
await service.cleanup_old_data(days=14)
|
||||||
except asyncio.CancelledError:
|
except asyncio.CancelledError:
|
||||||
break
|
break
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
@@ -214,9 +214,7 @@ async def mark_generation_nsfw(
|
|||||||
if not is_member:
|
if not is_member:
|
||||||
raise HTTPException(status_code=403, detail="Access denied")
|
raise HTTPException(status_code=403, detail="Access denied")
|
||||||
|
|
||||||
if not await generation_service.dao.generations.mark_nsfw(generation_id, request.is_nsfw):
|
await generation_service.dao.generations.mark_nsfw(generation_id, request.is_nsfw)
|
||||||
raise HTTPException(status_code=404, detail="Generation not found or already in the requested state")
|
|
||||||
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -2,6 +2,6 @@ from .AssetDTO import AssetResponse, AssetsResponse
|
|||||||
from .CharacterDTO import CharacterCreateRequest, CharacterUpdateRequest
|
from .CharacterDTO import CharacterCreateRequest, CharacterUpdateRequest
|
||||||
from .ExternalGenerationDTO import ExternalGenerationRequest
|
from .ExternalGenerationDTO import ExternalGenerationRequest
|
||||||
from .FinancialUsageDTO import FinancialReport, UsageStats, UsageByEntity
|
from .FinancialUsageDTO import FinancialReport, UsageStats, UsageByEntity
|
||||||
from .GenerationRequest import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse, PromptRequest, PromptResponse
|
from .GenerationRequest import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse, PromptRequest, PromptResponse, NsfwRequest
|
||||||
from .IdeaRequest import IdeaCreateRequest, IdeaUpdateRequest, IdeaResponse
|
from .IdeaRequest import IdeaCreateRequest, IdeaUpdateRequest, IdeaResponse
|
||||||
from .PostRequest import PostCreateRequest, PostUpdateRequest, AddGenerationsRequest
|
from .PostRequest import PostCreateRequest, PostUpdateRequest, AddGenerationsRequest
|
||||||
|
|||||||
Reference in New Issue
Block a user