Falko Schindler пре 1 година
родитељ
комит
c3f1854cd0

+ 2 - 1
nicegui/elements/leaflet.py

@@ -1,12 +1,13 @@
 from typing import Any, List, Tuple, cast
 
-from .. import binding, globals
+from .. import binding, globals  # pylint: disable=redefined-builtin
 from ..element import Element
 from ..events import GenericEventArguments
 from .leaflet_layer import Layer
 
 
 class Leaflet(Element, component='leaflet.js'):
+    # pylint: disable=import-outside-toplevel
     from .leaflet_layers import Marker as marker
     from .leaflet_layers import TileLayer as tile_layer
 

+ 5 - 2
nicegui/elements/leaflet_layer.py

@@ -1,8 +1,10 @@
+from __future__ import annotations
+
 from abc import abstractmethod
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, ClassVar, Optional
 
-from ..helpers import KWONLY_SLOTS
+from ..dataclasses import KWONLY_SLOTS
 
 if TYPE_CHECKING:
     from .leaflet import Leaflet
@@ -11,9 +13,10 @@ if TYPE_CHECKING:
 @dataclass(**KWONLY_SLOTS)
 class Layer:
     current_leaflet: ClassVar[Optional['Leaflet']] = None
-    leaflet: 'Leaflet' = field(init=False)
+    leaflet: Leaflet = field(init=False)
 
     def __post_init__(self) -> None:
+        assert self.current_leaflet is not None
         self.leaflet = self.current_leaflet
         self.leaflet.layers.append(self)
         self.leaflet.run_method('add_layer', self.to_dict())

+ 1 - 1
nicegui/elements/leaflet_layers.py

@@ -3,7 +3,7 @@ from typing import Dict, Tuple
 
 from typing_extensions import Self
 
-from ..helpers import KWONLY_SLOTS
+from ..dataclasses import KWONLY_SLOTS
 from .leaflet_layer import Layer