4d567b7ec1
* ruff-format: unify Black with Ruff * ruff lint. * v0.1.0 * run precommit * hand fixing * fix `not isinstance(...)` Co-authored-by: Thomas Brandého <thomas.brandeho@gmail.com> * rev isinstance(...) with noqa * exclude = ["*.pyi"] * apply precommit * apply format * revert pyi * fixing * Fix or Co-authored-by: Thomas Brandého <thomas.brandeho@gmail.com> * Update reflex/components/datadisplay/code.pyi * Apply suggestions from code review Co-authored-by: Thomas Brandého <thomas.brandeho@gmail.com> * Update tests/components/core/test_colors.py * Update reflex/app.py * Update reflex/app.py * Update benchmarks/test_benchmark_compile_pages.py * Update benchmarks/test_benchmark_compile_pages.py * Update benchmarks/test_benchmark_compile_pages.py --------- Co-authored-by: Thomas Brandého <thomas.brandeho@gmail.com>
94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
"""Import operations."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections import defaultdict
|
|
from typing import Dict, List, Optional
|
|
|
|
from reflex.base import Base
|
|
|
|
|
|
def merge_imports(*imports) -> ImportDict:
|
|
"""Merge multiple import dicts together.
|
|
|
|
Args:
|
|
*imports: The list of import dicts to merge.
|
|
|
|
Returns:
|
|
The merged import dicts.
|
|
"""
|
|
all_imports = defaultdict(list)
|
|
for import_dict in imports:
|
|
for lib, fields in import_dict.items():
|
|
all_imports[lib].extend(fields)
|
|
return all_imports
|
|
|
|
|
|
def collapse_imports(imports: ImportDict) -> ImportDict:
|
|
"""Remove all duplicate ImportVar within an ImportDict.
|
|
|
|
Args:
|
|
imports: The import dict to collapse.
|
|
|
|
Returns:
|
|
The collapsed import dict.
|
|
"""
|
|
return {lib: list(set(import_vars)) for lib, import_vars in imports.items()}
|
|
|
|
|
|
class ImportVar(Base):
|
|
"""An import var."""
|
|
|
|
# The name of the import tag.
|
|
tag: Optional[str]
|
|
|
|
# whether the import is default or named.
|
|
is_default: Optional[bool] = False
|
|
|
|
# The tag alias.
|
|
alias: Optional[str] = None
|
|
|
|
# Whether this import need to install the associated lib
|
|
install: Optional[bool] = True
|
|
|
|
# whether this import should be rendered or not
|
|
render: Optional[bool] = True
|
|
|
|
# whether this import package should be added to transpilePackages in next.config.js
|
|
# https://nextjs.org/docs/app/api-reference/next-config-js/transpilePackages
|
|
transpile: Optional[bool] = False
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""The name of the import.
|
|
|
|
Returns:
|
|
The name(tag name with alias) of tag.
|
|
"""
|
|
if self.alias:
|
|
return (
|
|
self.alias if self.is_default else " as ".join([self.tag, self.alias]) # type: ignore
|
|
)
|
|
else:
|
|
return self.tag or ""
|
|
|
|
def __hash__(self) -> int:
|
|
"""Define a hash function for the import var.
|
|
|
|
Returns:
|
|
The hash of the var.
|
|
"""
|
|
return hash(
|
|
(
|
|
self.tag,
|
|
self.is_default,
|
|
self.alias,
|
|
self.install,
|
|
self.render,
|
|
self.transpile,
|
|
)
|
|
)
|
|
|
|
|
|
ImportDict = Dict[str, List[ImportVar]]
|