Skip to content

Commit

Permalink
chore: update black to 24.1.0 (reata#569)
Browse files Browse the repository at this point in the history
  • Loading branch information
reata authored and liuzhou committed Feb 26, 2024
1 parent a412634 commit 1dfb12e
Show file tree
Hide file tree
Showing 5 changed files with 24 additions and 15 deletions.
6 changes: 3 additions & 3 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
@@ -1,15 +1,15 @@
repos:
- repo: https://github.com/psf/black
rev: 23.9.1
rev: 24.1.0
hooks:
- id: black
language_version: python3.8
- repo: https://github.com/pycqa/flake8.git
rev: 6.1.0
rev: 7.0.0
hooks:
- id: flake8
- repo: https://github.com/pre-commit/mirrors-mypy
rev: v1.5.1
rev: v1.8.0
hooks:
- id: mypy
additional_dependencies:
Expand Down
8 changes: 5 additions & 3 deletions sqllineage/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,9 +106,11 @@ def main(args=None) -> None:
runner = LineageRunner(
sql,
dialect=args.dialect,
metadata_provider=SQLAlchemyMetaDataProvider(args.sqlalchemy_url)
if args.sqlalchemy_url
else DummyMetaDataProvider(),
metadata_provider=(
SQLAlchemyMetaDataProvider(args.sqlalchemy_url)
if args.sqlalchemy_url
else DummyMetaDataProvider()
),
verbose=args.verbose,
draw_options={
"host": args.host,
Expand Down
8 changes: 5 additions & 3 deletions sqllineage/core/parser/sqlfluff/extractors/merge.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,9 +109,11 @@ def extract(
next_segment = segments[i + 1]
direct_source = SqlFluffSubQuery.of(
extract_innermost_bracketed(segment),
extract_identifier(next_segment)
if next_segment.type == "alias_expression"
else None,
(
extract_identifier(next_segment)
if next_segment.type == "alias_expression"
else None
),
)
holder.add_read(direct_source)

Expand Down
9 changes: 6 additions & 3 deletions sqllineage/core/parser/sqlfluff/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
extract_x: other pattern
first parameter of each function must be sqlfluff BaseSegment
"""

from typing import List, Optional, Tuple

from sqlfluff.core.parser import BaseSegment
Expand Down Expand Up @@ -156,9 +157,11 @@ def list_subqueries(segment: BaseSegment) -> List[SubQueryTuple]:
if is_subquery(target):
subquery = [
SubQueryTuple(
extract_innermost_bracketed(target)
if not is_set_expression(target)
else target,
(
extract_innermost_bracketed(target)
if not is_set_expression(target)
else target
),
extract_identifier(as_segment) if as_segment else None,
)
]
Expand Down
8 changes: 5 additions & 3 deletions sqllineage/io.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,11 @@ def to_cytoscape(graph: DiGraph, compound=False) -> List[Dict[str, Dict[str, Any
parents_dict = {
node.parent: {
"name": str(node.parent) if node.parent is not None else "<unknown>",
"type": type(node.parent).__name__
if node.parent is not None
else "Table or SubQuery",
"type": (
type(node.parent).__name__
if node.parent is not None
else "Table or SubQuery"
),
}
for node in graph.nodes
}
Expand Down

0 comments on commit 1dfb12e

Please sign in to comment.