karrio-server-graph 2025.5__py3-none-any.whl

karrio/server/graph/__init__.py

@@ -0,0 +1 @@
+__path__ = __import__('pkgutil').extend_path(__path__, __name__)  # type: ignore

karrio/server/graph/admin.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

karrio/server/graph/apps.py

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class GraphConfig(AppConfig):
+    name = 'karrio.server.graph'

karrio/server/graph/forms.py

@@ -0,0 +1,57 @@
+import django.forms as forms
+import django.contrib.auth.forms as auth
+import django.core.exceptions as exceptions
+import django.contrib.auth.tokens as tokens
+import django_email_verification.confirm as confirm
+
+import karrio.server.conf as conf
+import karrio.server.user.forms as user_forms
+from karrio.server.core.logging import logger
+
+
+class UserRegistrationForm(user_forms.SignUpForm):
+    pass
+
+
+class PasswordChangeForm(auth.PasswordChangeForm):
+    def __init__(self, user=None, *args, **kwargs):
+        super().__init__(user, *args, **kwargs)
+
+
+class ConfirmPasswordResetForm(auth.SetPasswordForm):
+    uid = forms.CharField(required=True, max_length=100)
+    token = forms.CharField(required=True, max_length=100)
+
+    def __init__(self, user=None, *args, **kwargs):
+        super().__init__(user, *args, **kwargs)
+
+    def clean_token(self):
+        token = self.cleaned_data["token"]
+        valid_link = tokens.default_token_generator.check_token(self.user, token)
+
+        if not valid_link:
+            raise exceptions.ValidationError("invalid or expired url token")
+
+
+class ResetPasswordRequestForm(auth.PasswordResetForm):
+    redirect_url = forms.URLField()
+    from_email = confirm._get_validated_field("EMAIL_FROM_ADDRESS")
+
+    def save(self, **kwargs):
+        try:
+            super().save(
+                **{
+                    **kwargs,
+                    "extra_email_context": dict(
+                        app_name=conf.settings.app_name,
+                        redirect_url=self.cleaned_data["redirect_url"],
+                    ),
+                    "email_template_name": "karrio/password_reset_email.html",
+                    "from_email": self.from_email,
+                }
+            )
+        except Exception as e:
+            logger.error("Password reset email failed", error=str(e))
+            raise exceptions.ValidationError(
+                "An error occurred while sending the email"
+            )

karrio/server/graph/management/commands/export_schema.py

@@ -0,0 +1,9 @@
+from django.core.management import BaseCommand
+from strawberry.printer import print_schema
+
+from karrio.server.graph.schema import schema
+
+class Command(BaseCommand):
+    help = 'Exports the strawberry graphql schema'
+    def handle(self, *args, **options):
+        print(print_schema(schema))

karrio/server/graph/migrations/0001_initial.py

@@ -0,0 +1,37 @@
+# Generated by Django 3.1.7 on 2021-03-26 14:53
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+import karrio.server.core.models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('manager', '0009_auto_20210326_1425'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Template',
+            fields=[
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('id', models.CharField(default=karrio.server.core.models.uuid, editable=False, max_length=50, primary_key=True, serialize=False)),
+                ('label', models.CharField(max_length=50)),
+                ('is_default', models.BooleanField(null=True)),
+                ('address', models.OneToOneField(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to='manager.address')),
+                ('created_by', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+                ('customs', models.OneToOneField(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to='manager.customs')),
+                ('parcel', models.OneToOneField(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to='manager.parcel')),
+            ],
+            options={
+                'db_table': 'template',
+                'ordering': ['-created_at'],
+            },
+        ),
+    ]

karrio/server/graph/migrations/0002_auto_20210512_1353.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.2.2 on 2021-05-12 13:53
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('graph', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='template',
+            options={'ordering': ['-is_default', '-created_by']},
+        ),
+        migrations.AlterField(
+            model_name='template',
+            name='is_default',
+            field=models.BooleanField(blank=True, default=False),
+        ),
+    ]

karrio/server/graph/models.py

@@ -0,0 +1,44 @@
+from django.conf import settings
+from django.db import models
+
+from karrio.server.core.models import OwnedEntity, uuid, register_model
+from karrio.server.manager.models import Customs, Parcel, Address
+
+
+@register_model
+class Template(OwnedEntity):
+    HIDDEN_PROPS = (*(("org",) if settings.MULTI_ORGANIZATIONS else tuple()),)
+
+    class Meta:
+        db_table = "template"
+        ordering = ["-is_default", "-created_by"]
+
+    id = models.CharField(max_length=50, primary_key=True, default=uuid, editable=False)
+    label = models.CharField(max_length=50)
+    is_default = models.BooleanField(blank=True, default=False)
+
+    address = models.OneToOneField(
+        Address, on_delete=models.CASCADE, null=True, blank=True
+    )
+    customs = models.OneToOneField(
+        Customs, on_delete=models.CASCADE, null=True, blank=True
+    )
+    parcel = models.OneToOneField(
+        Parcel, on_delete=models.CASCADE, null=True, blank=True
+    )
+
+    def delete(self, *args, **kwargs):
+        attachment = next(
+            (
+                entity
+                for entity in [self.address, self.customs, self.parcel]
+                if entity is not None
+            ),
+            super(),
+        )
+
+        return attachment.delete(*args, **kwargs)
+
+    @property
+    def object_type(self):
+        return "template"

karrio/server/graph/schema.py

@@ -0,0 +1,44 @@
+import pkgutil
+import strawberry
+import strawberry
+import strawberry.schema.config as config
+
+from karrio.server.core.logging import logger
+import karrio.server.graph.schemas as schemas
+
+QUERIES: list = []
+MUTATIONS: list = []
+EXTRA_TYPES: list = []
+
+# Register karrio graphql schemas
+for _, name, _ in pkgutil.iter_modules(schemas.__path__):  # type: ignore
+    try:
+        schema = __import__(f"{schemas.__name__}.{name}", fromlist=[name])
+        if hasattr(schema, "Query"):
+            QUERIES.append(schema.Query)
+        if hasattr(schema, "Mutation"):
+            MUTATIONS.append(schema.Mutation)
+        if hasattr(schema, "extra_types"):
+            EXTRA_TYPES += schema.extra_types
+    except Exception as e:
+        logger.warning("Failed to register GraphQL schema", schema_name=name, error=str(e))
+        logger.exception("GraphQL schema registration error", schema_name=name)
+
+
+@strawberry.type
+class Query(*QUERIES):  # type: ignore
+    pass
+
+
+@strawberry.type
+class Mutation(*MUTATIONS):  # type: ignore
+    pass
+
+
+schema = strawberry.Schema(  # type: ignore
+    query=Query,
+    mutation=Mutation,
+    types=[*EXTRA_TYPES],
+    config=config.StrawberryConfig(auto_camel_case=False),
+    extensions=[],
+)

karrio/server/graph/schemas/__init__.py

@@ -0,0 +1,2 @@
+# type: ignore
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)

karrio/server/graph/schemas/base/__init__.py

@@ -0,0 +1,385 @@
+import typing
+import strawberry
+from strawberry.types import Info
+
+import karrio.server.core.models as core
+import karrio.server.graph.models as graph
+import karrio.server.manager.models as manager
+import karrio.server.providers.models as providers
+import karrio.server.manager.serializers as manager_serializers
+import karrio.server.graph.schemas.base.mutations as mutations
+import karrio.server.graph.schemas.base.inputs as inputs
+import karrio.server.graph.schemas.base.types as types
+import karrio.server.graph.utils as utils
+
+# extra_types = [*types.CarrierSettings.values()]
+extra_types = []
+
+
+@strawberry.type
+class Query:
+    user: types.UserType = strawberry.field(resolver=types.UserType.resolve)
+    token: types.TokenType = strawberry.field(resolver=types.TokenType.resolve)
+    api_keys: typing.List[types.APIKeyType] = strawberry.field(
+        resolver=types.APIKeyType.resolve_list
+    )
+    workspace_config: typing.Optional[types.WorkspaceConfigType] = strawberry.field(
+        resolver=types.WorkspaceConfigType.resolve
+    )
+    system_usage: types.SystemUsageType = strawberry.field(
+        resolver=types.SystemUsageType.resolve
+    )
+
+    user_connections: utils.Connection[types.CarrierConnectionType] = strawberry.field(
+        resolver=types.CarrierConnectionType.resolve_list
+    )
+    system_connections: utils.Connection[types.SystemConnectionType] = strawberry.field(
+        resolver=types.SystemConnectionType.resolve_list
+    )
+
+    default_templates: types.DefaultTemplatesType = strawberry.field(
+        resolver=types.DefaultTemplatesType.resolve
+    )
+    address_templates: utils.Connection[types.AddressTemplateType] = strawberry.field(
+        resolver=types.AddressTemplateType.resolve_list
+    )
+    customs_templates: utils.Connection[types.CustomsTemplateType] = strawberry.field(
+        resolver=types.CustomsTemplateType.resolve_list
+    )
+    parcel_templates: utils.Connection[types.ParcelTemplateType] = strawberry.field(
+        resolver=types.ParcelTemplateType.resolve_list
+    )
+
+    log: typing.Optional[types.LogType] = strawberry.field(
+        resolver=types.LogType.resolve
+    )
+    logs: utils.Connection[types.LogType] = strawberry.field(
+        resolver=types.LogType.resolve_list
+    )
+
+    tracing_record: typing.Optional[types.TracingRecordType] = strawberry.field(
+        resolver=types.TracingRecordType.resolve
+    )
+    tracing_records: utils.Connection[types.TracingRecordType] = strawberry.field(
+        resolver=types.TracingRecordType.resolve_list
+    )
+
+    shipment: typing.Optional[types.ShipmentType] = strawberry.field(
+        resolver=types.ShipmentType.resolve
+    )
+    shipments: utils.Connection[types.ShipmentType] = strawberry.field(
+        resolver=types.ShipmentType.resolve_list
+    )
+
+    tracker: typing.Optional[types.TrackerType] = strawberry.field(
+        resolver=types.TrackerType.resolve
+    )
+    trackers: utils.Connection[types.TrackerType] = strawberry.field(
+        resolver=types.TrackerType.resolve_list
+    )
+
+    rate_sheet: typing.Optional[types.RateSheetType] = strawberry.field(
+        resolver=types.RateSheetType.resolve
+    )
+    rate_sheets: utils.Connection[types.RateSheetType] = strawberry.field(
+        resolver=types.RateSheetType.resolve_list
+    )
+
+    manifest: typing.Optional[types.ManifestType] = strawberry.field(
+        resolver=types.ManifestType.resolve
+    )
+    manifests: utils.Connection[types.ManifestType] = strawberry.field(
+        resolver=types.ManifestType.resolve_list
+    )
+
+    carrier_connection: typing.Optional[types.CarrierConnectionType] = strawberry.field(
+        resolver=types.CarrierConnectionType.resolve
+    )
+    carrier_connections: utils.Connection[types.CarrierConnectionType] = (
+        strawberry.field(resolver=types.CarrierConnectionType.resolve_list)
+    )
+
+    metafield: typing.Optional[types.MetafieldType] = strawberry.field(
+        resolver=types.MetafieldType.resolve
+    )
+    metafields: utils.Connection[types.MetafieldType] = strawberry.field(
+        resolver=types.MetafieldType.resolve_list
+    )
+
+
+@strawberry.type
+class Mutation:
+    @strawberry.mutation
+    def update_user(
+        self, info: Info, input: inputs.UpdateUserInput
+    ) -> mutations.UserUpdateMutation:
+        return mutations.UserUpdateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def register_user(
+        self, info: Info, input: inputs.RegisterUserMutationInput
+    ) -> mutations.RegisterUserMutation:
+        return mutations.RegisterUserMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_workspace_config(
+        self, info: Info, input: inputs.WorkspaceConfigMutationInput
+    ) -> mutations.WorkspaceConfigMutation:
+        return mutations.WorkspaceConfigMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def mutate_token(
+        self, info: Info, input: inputs.TokenMutationInput
+    ) -> mutations.TokenMutation:
+        return mutations.TokenMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def create_api_key(
+        self, info: Info, input: inputs.CreateAPIKeyMutationInput
+    ) -> mutations.CreateAPIKeyMutation:
+        return mutations.CreateAPIKeyMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_api_key(
+        self, info: Info, input: inputs.DeleteAPIKeyMutationInput
+    ) -> mutations.DeleteAPIKeyMutation:
+        return mutations.DeleteAPIKeyMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def request_email_change(
+        self, info: Info, input: inputs.RequestEmailChangeMutationInput
+    ) -> mutations.RequestEmailChangeMutation:
+        return mutations.RequestEmailChangeMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def confirm_email_change(
+        self, info: Info, input: inputs.ConfirmEmailChangeMutationInput
+    ) -> mutations.ConfirmEmailChangeMutation:
+        return mutations.ConfirmEmailChangeMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def confirm_email(
+        self, info: Info, input: inputs.ConfirmEmailMutationInput
+    ) -> mutations.ConfirmEmailMutation:
+        return mutations.ConfirmEmailMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def change_password(
+        self, info: Info, input: inputs.ChangePasswordMutationInput
+    ) -> mutations.ChangePasswordMutation:
+        return mutations.ChangePasswordMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def request_password_reset(
+        self, info: Info, input: inputs.RequestPasswordResetMutationInput
+    ) -> mutations.RequestPasswordResetMutation:
+        return mutations.RequestPasswordResetMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def confirm_password_reset(
+        self, info: Info, input: inputs.ConfirmPasswordResetMutationInput
+    ) -> mutations.ConfirmPasswordResetMutation:
+        return mutations.ConfirmPasswordResetMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def enable_multi_factor(
+        self, info: Info, input: inputs.EnableMultiFactorMutationInput
+    ) -> mutations.EnableMultiFactorMutation:
+        return mutations.EnableMultiFactorMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def confirm_multi_factor(
+        self, info: Info, input: inputs.ConfirmMultiFactorMutationInput
+    ) -> mutations.ConfirmMultiFactorMutation:
+        return mutations.ConfirmMultiFactorMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def disable_multi_factor(
+        self, info: Info, input: inputs.DisableMultiFactorMutationInput
+    ) -> mutations.DisableMultiFactorMutation:
+        return mutations.DisableMultiFactorMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def create_address_template(
+        self, info: Info, input: inputs.CreateAddressTemplateInput
+    ) -> mutations.CreateAddressTemplateMutation:
+        return mutations.CreateAddressTemplateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_address_template(
+        self, info: Info, input: inputs.UpdateAddressTemplateInput
+    ) -> mutations.UpdateAddressTemplateMutation:
+        return mutations.UpdateAddressTemplateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def create_customs_template(
+        self, info: Info, input: inputs.CreateCustomsTemplateInput
+    ) -> mutations.CreateCustomsTemplateMutation:
+        return mutations.CreateCustomsTemplateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_customs_template(
+        self, info: Info, input: inputs.UpdateCustomsTemplateInput
+    ) -> mutations.UpdateCustomsTemplateMutation:
+        return mutations.UpdateCustomsTemplateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def create_parcel_template(
+        self, info: Info, input: inputs.CreateParcelTemplateInput
+    ) -> mutations.CreateParcelTemplateMutation:
+        return mutations.CreateParcelTemplateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_parcel_template(
+        self, info: Info, input: inputs.UpdateParcelTemplateInput
+    ) -> mutations.UpdateParcelTemplateMutation:
+        return mutations.UpdateParcelTemplateMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def create_carrier_connection(
+        self, info: Info, input: inputs.CreateCarrierConnectionMutationInput
+    ) -> mutations.CreateCarrierConnectionMutation:
+        return mutations.CreateCarrierConnectionMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_carrier_connection(
+        self, info: Info, input: inputs.UpdateCarrierConnectionMutationInput
+    ) -> mutations.UpdateCarrierConnectionMutation:
+        return mutations.UpdateCarrierConnectionMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def mutate_system_connection(
+        self, info: Info, input: inputs.SystemCarrierMutationInput
+    ) -> mutations.SystemCarrierMutation:
+        return mutations.SystemCarrierMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_carrier_connection(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info, model=providers.Carrier, **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def partial_shipment_update(
+        self, info: Info, input: inputs.PartialShipmentMutationInput
+    ) -> mutations.PartialShipmentMutation:
+        return mutations.PartialShipmentMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def mutate_metadata(
+        self, info: Info, input: inputs.MetadataMutationInput
+    ) -> mutations.MetadataMutation:
+        return mutations.MetadataMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def change_shipment_status(
+        self, info: Info, input: inputs.ChangeShipmentStatusMutationInput
+    ) -> mutations.ChangeShipmentStatusMutation:
+        return mutations.ChangeShipmentStatusMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_template(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info, model=graph.Template, **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def discard_commodity(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info,
+            model=manager.Commodity,
+            validator=manager_serializers.can_mutate_commodity,
+            **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def discard_customs(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info,
+            model=manager.Customs,
+            validator=manager_serializers.can_mutate_customs,
+            **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def discard_parcel(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info,
+            model=manager.Parcel,
+            validator=manager_serializers.can_mutate_parcel,
+            **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def create_rate_sheet(
+        self, info: Info, input: inputs.CreateRateSheetMutationInput
+    ) -> mutations.CreateRateSheetMutation:
+        return mutations.CreateRateSheetMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_rate_sheet(
+        self, info: Info, input: inputs.UpdateRateSheetMutationInput
+    ) -> mutations.UpdateRateSheetMutation:
+        return mutations.UpdateRateSheetMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_rate_sheet_zone_cell(
+        self, info: Info, input: inputs.UpdateRateSheetZoneCellMutationInput
+    ) -> mutations.UpdateRateSheetZoneCellMutation:
+        return mutations.UpdateRateSheetZoneCellMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def batch_update_rate_sheet_cells(
+        self, info: Info, input: inputs.BatchUpdateRateSheetCellsMutationInput
+    ) -> mutations.BatchUpdateRateSheetCellsMutation:
+        return mutations.BatchUpdateRateSheetCellsMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_rate_sheet_service(
+        self, info: Info, input: inputs.DeleteRateSheetServiceMutationInput
+    ) -> mutations.DeleteRateSheetServiceMutation:
+        return mutations.DeleteRateSheetServiceMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_rate_sheet(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info, model=providers.RateSheet, **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def create_metafield(
+        self, info: Info, input: inputs.CreateMetafieldInput
+    ) -> mutations.CreateMetafieldMutation:
+        return mutations.CreateMetafieldMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def update_metafield(
+        self, info: Info, input: inputs.UpdateMetafieldInput
+    ) -> mutations.UpdateMetafieldMutation:
+        return mutations.UpdateMetafieldMutation.mutate(info, **input.to_dict())
+
+    @strawberry.mutation
+    def delete_metafield(
+        self, info: Info, input: inputs.DeleteMutationInput
+    ) -> mutations.DeleteMutation:
+        return mutations.DeleteMutation.mutate(
+            info, model=core.Metafield, **input.to_dict()
+        )
+
+    @strawberry.mutation
+    def update_service_zone(
+        self, info: Info, input: inputs.UpdateServiceZoneMutationInput
+    ) -> mutations.UpdateServiceZoneMutation:
+        return mutations.UpdateServiceZoneMutation.mutate(info, **input.to_dict())