From 87dfbad683adcd2ac6c0f0648663af1179d1f100 Mon Sep 17 00:00:00 2001
From: Martin Mares <mj@ucw.cz>
Date: Fri, 11 Nov 2022 19:23:25 +0100
Subject: [PATCH] =?UTF-8?q?mo.web.table:=20Argumenty=20requestu=20mohou=20?=
 =?UTF-8?q?b=C3=BDt=20jak=20ImmutableMultiDict,=20tak=20MultiDict?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mo/web/table.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/mo/web/table.py b/mo/web/table.py
index 65fdde48..bc6975f5 100644
--- a/mo/web/table.py
+++ b/mo/web/table.py
@@ -6,7 +6,7 @@ import io
 from markupsafe import Markup
 from typing import Any, Dict, List, Sequence, Optional, Iterable, Union
 import urllib.parse
-from werkzeug.datastructures import ImmutableMultiDict
+from werkzeug.datastructures import MultiDict, ImmutableMultiDict
 import werkzeug.exceptions
 
 from mo.csv import FileFormat
@@ -206,7 +206,7 @@ class Table:
 
         return Markup("\n".join(tab))
 
-    def get_columns_checkboxes(self, line_prefix: str = "", args: Optional[ImmutableMultiDict] = None) -> Markup:
+    def get_columns_checkboxes(self, line_prefix: str = "", args: Union[None, MultiDict, ImmutableMultiDict] = None) -> Markup:
         out = [line_prefix + '<input type="hidden" name="do_column_selection" value="1">']
         for c in self.columns:
             if c.in_export is None:
@@ -251,7 +251,7 @@ class Table:
 
         yield out.getvalue().encode(fmt.get_charset())
 
-    def send_as(self, format: Union[FileFormat, str], streaming: bool = False, args: Optional[ImmutableMultiDict] = None) -> Response:
+    def send_as(self, format: Union[FileFormat, str], streaming: bool = False, args: Union[None, MultiDict, ImmutableMultiDict] = None) -> Response:
         try:
             fmt = FileFormat.coerce(format)
         except ValueError:
-- 
GitLab