mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
from documents.serialisers import CorrespondentField
|
|
from documents.serialisers import DocumentTypeField
|
|
from documents.serialisers import TagsField
|
|
from paperless_mail.models import MailAccount
|
|
from paperless_mail.models import MailRule
|
|
from rest_framework import serializers
|
|
|
|
|
|
class ObfuscatedPasswordField(serializers.Field):
|
|
"""
|
|
Sends *** string instead of password in the clear
|
|
"""
|
|
|
|
def to_representation(self, value):
|
|
return "*" * len(value)
|
|
|
|
def to_internal_value(self, data):
|
|
return data
|
|
|
|
|
|
class MailAccountSerializer(serializers.ModelSerializer):
|
|
password = ObfuscatedPasswordField()
|
|
|
|
class Meta:
|
|
model = MailAccount
|
|
depth = 1
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"imap_server",
|
|
"imap_port",
|
|
"imap_security",
|
|
"username",
|
|
"password",
|
|
"character_set",
|
|
]
|
|
|
|
def update(self, instance, validated_data):
|
|
if "password" in validated_data:
|
|
if len(validated_data.get("password").replace("*", "")) == 0:
|
|
validated_data.pop("password")
|
|
super().update(instance, validated_data)
|
|
return instance
|
|
|
|
def create(self, validated_data):
|
|
mail_account = MailAccount.objects.create(**validated_data)
|
|
return mail_account
|
|
|
|
|
|
class AccountField(serializers.PrimaryKeyRelatedField):
|
|
def get_queryset(self):
|
|
return MailAccount.objects.all().order_by("-id")
|
|
|
|
|
|
class MailRuleSerializer(serializers.ModelSerializer):
|
|
account = AccountField(required=True)
|
|
action_parameter = serializers.CharField(
|
|
allow_null=True,
|
|
required=False,
|
|
default="",
|
|
)
|
|
assign_correspondent = CorrespondentField(allow_null=True, required=False)
|
|
assign_tags = TagsField(many=True, allow_null=True, required=False)
|
|
assign_document_type = DocumentTypeField(allow_null=True, required=False)
|
|
order = serializers.IntegerField(required=False)
|
|
|
|
class Meta:
|
|
model = MailRule
|
|
depth = 1
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"account",
|
|
"folder",
|
|
"filter_from",
|
|
"filter_subject",
|
|
"filter_body",
|
|
"filter_attachment_filename",
|
|
"maximum_age",
|
|
"action",
|
|
"action_parameter",
|
|
"assign_title_from",
|
|
"assign_tags",
|
|
"assign_correspondent_from",
|
|
"assign_correspondent",
|
|
"assign_document_type",
|
|
"order",
|
|
"attachment_type",
|
|
]
|
|
|
|
def update(self, instance, validated_data):
|
|
super().update(instance, validated_data)
|
|
return instance
|
|
|
|
def create(self, validated_data):
|
|
if "assign_tags" in validated_data:
|
|
assign_tags = validated_data.pop("assign_tags")
|
|
mail_rule = MailRule.objects.create(**validated_data)
|
|
if assign_tags:
|
|
mail_rule.assign_tags.set(assign_tags)
|
|
return mail_rule
|
|
|
|
def validate(self, attrs):
|
|
if (
|
|
attrs["action"] == MailRule.MailAction.TAG
|
|
or attrs["action"] == MailRule.MailAction.MOVE
|
|
) and attrs["action_parameter"] is None:
|
|
raise serializers.ValidationError("An action parameter is required.")
|
|
|
|
return attrs
|