From ab63ba1c541b03fd8b6b0a66ed321cb7d3def734 Mon Sep 17 00:00:00 2001
From: Sli <antoine@bartuccio.fr>
Date: Mon, 21 Oct 2024 13:26:11 +0200
Subject: [PATCH] Remove ajax_select from accounting

---
 .../components/ajax-select-index.ts           | 60 +++++++++++++++++++
 .../templates/accounting/operation_edit.jinja |  8 +--
 accounting/views.py                           | 45 +++++++++++---
 accounting/widgets/select.py                  | 45 ++++++++++++++
 4 files changed, 146 insertions(+), 12 deletions(-)
 create mode 100644 accounting/static/webpack/accounting/components/ajax-select-index.ts
 create mode 100644 accounting/widgets/select.py

diff --git a/accounting/static/webpack/accounting/components/ajax-select-index.ts b/accounting/static/webpack/accounting/components/ajax-select-index.ts
new file mode 100644
index 00000000..3fc93cf3
--- /dev/null
+++ b/accounting/static/webpack/accounting/components/ajax-select-index.ts
@@ -0,0 +1,60 @@
+import { AjaxSelect } from "#core:core/components/ajax-select-base";
+import { registerComponent } from "#core:utils/web-components";
+import type { TomOption } from "tom-select/dist/types/types";
+import type { escape_html } from "tom-select/dist/types/utils";
+import {
+  type ClubAccountSchema,
+  type CompanySchema,
+  accountingSearchClubAccount,
+  accountingSearchCompany,
+} from "#openapi";
+
+@registerComponent("club-account-ajax-select")
+export class ClubAccountAjaxSelect extends AjaxSelect {
+  protected valueField = "id";
+  protected labelField = "name";
+  protected searchField = ["code", "name"];
+
+  protected async search(query: string): Promise<TomOption[]> {
+    const resp = await accountingSearchClubAccount({ query: { search: query } });
+    if (resp.data) {
+      return resp.data.results;
+    }
+    return [];
+  }
+
+  protected renderOption(item: ClubAccountSchema, sanitize: typeof escape_html) {
+    return `<div class="select-item">
+            <span class="select-item-text">${sanitize(item.name)}</span>
+          </div>`;
+  }
+
+  protected renderItem(item: ClubAccountSchema, sanitize: typeof escape_html) {
+    return `<span>${sanitize(item.name)}</span>`;
+  }
+}
+
+@registerComponent("company-ajax-select")
+export class CompanyAjaxSelect extends AjaxSelect {
+  protected valueField = "id";
+  protected labelField = "name";
+  protected searchField = ["code", "name"];
+
+  protected async search(query: string): Promise<TomOption[]> {
+    const resp = await accountingSearchCompany({ query: { search: query } });
+    if (resp.data) {
+      return resp.data.results;
+    }
+    return [];
+  }
+
+  protected renderOption(item: CompanySchema, sanitize: typeof escape_html) {
+    return `<div class="select-item">
+            <span class="select-item-text">${sanitize(item.name)}</span>
+          </div>`;
+  }
+
+  protected renderItem(item: CompanySchema, sanitize: typeof escape_html) {
+    return `<span>${sanitize(item.name)}</span>`;
+  }
+}
diff --git a/accounting/templates/accounting/operation_edit.jinja b/accounting/templates/accounting/operation_edit.jinja
index e8c34364..4a75cb83 100644
--- a/accounting/templates/accounting/operation_edit.jinja
+++ b/accounting/templates/accounting/operation_edit.jinja
@@ -61,10 +61,10 @@
   <script>
     $( function() {
       var target_type = $('#id_target_type');
-      var user = $('#id_user_wrapper');
-      var club = $('#id_club_wrapper');
-      var club_account = $('#id_club_account_wrapper');
-      var company = $('#id_company_wrapper');
+      var user = $('user-ajax-select');
+      var club = $('club-ajax-select');
+      var club_account = $('club-account-ajax-select');
+      var company = $('company-ajax-select');
       var other = $('#id_target_label');
       var need_link = $('#id_need_link_full');
       function update_targets () {
diff --git a/accounting/views.py b/accounting/views.py
index dd1cc974..928dc009 100644
--- a/accounting/views.py
+++ b/accounting/views.py
@@ -15,7 +15,6 @@
 
 import collections
 
-from ajax_select.fields import AutoCompleteSelectField
 from django import forms
 from django.conf import settings
 from django.core.exceptions import PermissionDenied, ValidationError
@@ -39,6 +38,13 @@ from accounting.models import (
     Operation,
     SimplifiedAccountingType,
 )
+from accounting.widgets.select import (
+    AutoCompleteSelectClubAccount,
+    AutoCompleteSelectCompany,
+)
+from club.models import Club
+from club.widgets.select import AutoCompleteSelectClub
+from core.models import User
 from core.views import (
     CanCreateMixin,
     CanEditMixin,
@@ -47,6 +53,7 @@ from core.views import (
     TabedViewMixin,
 )
 from core.views.forms import SelectDate, SelectFile
+from core.views.widgets.select import AutoCompleteSelectUser
 from counter.models import Counter, Product, Selling
 
 # Main accounting view
@@ -334,12 +341,30 @@ class OperationForm(forms.ModelForm):
             "invoice": SelectFile,
         }
 
-    user = AutoCompleteSelectField("users", help_text=None, required=False)
-    club_account = AutoCompleteSelectField(
-        "club_accounts", help_text=None, required=False
+    user = forms.ModelChoiceField(
+        help_text=None,
+        required=False,
+        widget=AutoCompleteSelectUser,
+        queryset=User.objects.all(),
+    )
+    club_account = forms.ModelChoiceField(
+        help_text=None,
+        required=False,
+        widget=AutoCompleteSelectClubAccount,
+        queryset=ClubAccount.objects.all(),
+    )
+    club = forms.ModelChoiceField(
+        help_text=None,
+        required=False,
+        widget=AutoCompleteSelectClub,
+        queryset=Club.objects.all(),
+    )
+    company = forms.ModelChoiceField(
+        help_text=None,
+        required=False,
+        widget=AutoCompleteSelectCompany,
+        queryset=Company.objects.all(),
     )
-    club = AutoCompleteSelectField("clubs", help_text=None, required=False)
-    company = AutoCompleteSelectField("companies", help_text=None, required=False)
     need_link = forms.BooleanField(
         label=_("Link this operation to the target account"),
         required=False,
@@ -817,8 +842,12 @@ class LabelDeleteView(CanEditMixin, DeleteView):
 
 
 class CloseCustomerAccountForm(forms.Form):
-    user = AutoCompleteSelectField(
-        "users", label=_("Refound this account"), help_text=None, required=True
+    user = forms.ModelChoiceField(
+        label=_("Refound this account"),
+        help_text=None,
+        required=True,
+        widget=AutoCompleteSelectUser,
+        queryset=User.objects.all(),
     )
 
 
diff --git a/accounting/widgets/select.py b/accounting/widgets/select.py
new file mode 100644
index 00000000..b9d85e83
--- /dev/null
+++ b/accounting/widgets/select.py
@@ -0,0 +1,45 @@
+from django.forms import Select, SelectMultiple
+
+from accounting.models import ClubAccount, Company
+from accounting.schemas import ClubAccountSchema, CompanySchema
+from core.views.widgets.select import AutoCompleteSelectMixin
+
+
+class AutoCompleteSelectClubAccount(AutoCompleteSelectMixin, Select):
+    component_name = "club-account-ajax-select"
+    model = ClubAccount
+    schema = ClubAccountSchema
+
+    js = [
+        "webpack/accounting/components/ajax-select-index.ts",
+    ]
+
+
+class AutoCompleteSelectMultipleClubAccount(AutoCompleteSelectMixin, SelectMultiple):
+    component_name = "club-account-ajax-select"
+    model = ClubAccount
+    schema = ClubAccountSchema
+
+    js = [
+        "webpack/accounting/components/ajax-select-index.ts",
+    ]
+
+
+class AutoCompleteSelectCompany(AutoCompleteSelectMixin, Select):
+    component_name = "company-ajax-select"
+    model = Company
+    schema = CompanySchema
+
+    js = [
+        "webpack/accounting/components/ajax-select-index.ts",
+    ]
+
+
+class AutoCompleteSelectMultipleCompany(AutoCompleteSelectMixin, SelectMultiple):
+    component_name = "company-ajax-select"
+    model = Company
+    schema = CompanySchema
+
+    js = [
+        "webpack/accounting/components/ajax-select-index.ts",
+    ]