Merge branch 'main' into Rule_Special_Characters_In_Names

src/sqlfluff/__init__.py

@@ -3,14 +3,22 @@
 import pytest
 
 # Expose the public API.
-from sqlfluff.api import lint, fix, parse, list_rules, list_dialects  # noqa: F401
+from sqlfluff.api import lint, fix, parse, list_rules, list_dialects
 
 # Import metadata (using importlib_metadata backport for python versions <3.8)
 if sys.version_info < (3, 8, 0):
     import importlib_metadata as metadata
 else:
     from importlib import metadata
 
+__all__ = (
+    "lint",
+    "fix",
+    "parse",
+    "list_rules",
+    "list_dialects",
+)
+
 # Get the current version
 __version__ = metadata.version("sqlfluff")
 

src/sqlfluff/api/__init__.py

@@ -1,7 +1,14 @@
 """Elements which wrap the sqlfluff core library for public use."""
 
-# flake8: noqa: F401
-
 # Expose the simple api
 from sqlfluff.api.simple import lint, fix, parse, APIParsingError
 from sqlfluff.api.info import list_rules, list_dialects
+
+__all__ = (
+    "lint",
+    "fix",
+    "parse",
+    "APIParsingError",
+    "list_rules",
+    "list_dialects",
+)

src/sqlfluff/core/__init__.py

@@ -1,6 +1,5 @@
 """The core elements of sqlfluff."""
 
-# flake8: noqa: F401
 import tblib.pickling_support  # type: ignore
 
 # Config objects
@@ -26,6 +25,21 @@
 # Timing objects
 from sqlfluff.core.timing import TimingSummary
 
+__all__ = (
+    "FluffConfig",
+    "Linter",
+    "Lexer",
+    "Parser",
+    "dialect_selector",
+    "dialect_readout",
+    "SQLBaseError",
+    "SQLTemplaterError",
+    "SQLLexError",
+    "SQLParseError",
+    "SQLLintError",
+    "SQLFluffUserError",
+    "TimingSummary",
+)
 
 # This is for "sqlfluff lint" and "sqlfluff fix" multiprocessing (--processes)
 # support. If an exception (i.e. runtime error) occurs in a worker process, we

src/sqlfluff/core/linter/__init__.py

@@ -1,8 +1,15 @@
 """Linter class and helper classes."""
 
-# flake8: noqa: F401
-
 from sqlfluff.core.linter.common import RuleTuple, ParsedString, NoQaDirective
 from sqlfluff.core.linter.linted_file import LintedFile
 from sqlfluff.core.linter.linting_result import LintingResult
 from sqlfluff.core.linter.linter import Linter
+
+__all__ = (
+    "RuleTuple",
+    "ParsedString",
+    "NoQaDirective",
+    "LintedFile",
+    "LintingResult",
+    "Linter",
+)

src/sqlfluff/core/parser/__init__.py

@@ -1,6 +1,4 @@
-""" init file for the parser """
-
-# flake8: noqa: F401
+"""init file for the parser."""
 
 from sqlfluff.core.parser.segments import (
     BaseSegment,
@@ -36,3 +34,40 @@
 from sqlfluff.core.parser.lexer import Lexer, StringLexer, RegexLexer
 from sqlfluff.core.parser.parser import Parser
 from sqlfluff.core.parser.matchable import Matchable
+
+__all__ = (
+    "BaseSegment",
+    "BaseFileSegment",
+    "RawSegment",
+    "CodeSegment",
+    "UnlexableSegment",
+    "CommentSegment",
+    "WhitespaceSegment",
+    "NewlineSegment",
+    "KeywordSegment",
+    "SymbolSegment",
+    "Indent",
+    "Dedent",
+    "SegmentGenerator",
+    "Sequence",
+    "GreedyUntil",
+    "StartsWith",
+    "OneOf",
+    "Delimited",
+    "Bracketed",
+    "AnyNumberOf",
+    "Ref",
+    "Anything",
+    "Nothing",
+    "OptionallyBracketed",
+    "Conditional",
+    "StringParser",
+    "NamedParser",
+    "RegexParser",
+    "PositionMarker",
+    "Lexer",
+    "StringLexer",
+    "RegexLexer",
+    "Parser",
+    "Matchable",
+)

src/sqlfluff/core/parser/grammar/__init__.py

@@ -1,10 +1,23 @@
 """Definitions of grammars."""
 
-# flake8: noqa: F401
-
 from sqlfluff.core.parser.grammar.base import Ref, Anything, Nothing
 from sqlfluff.core.parser.grammar.anyof import AnyNumberOf, OneOf, OptionallyBracketed
 from sqlfluff.core.parser.grammar.delimited import Delimited
 from sqlfluff.core.parser.grammar.greedy import GreedyUntil, StartsWith
 from sqlfluff.core.parser.grammar.sequence import Sequence, Bracketed
 from sqlfluff.core.parser.grammar.conditional import Conditional
+
+__all__ = (
+    "Ref",
+    "Anything",
+    "Nothing",
+    "AnyNumberOf",
+    "OneOf",
+    "OptionallyBracketed",
+    "Delimited",
+    "GreedyUntil",
+    "StartsWith",
+    "Sequence",
+    "Bracketed",
+    "Conditional",
+)

src/sqlfluff/core/parser/segments/__init__.py

@@ -1,6 +1,5 @@
 """Definitions of the segment classes."""
 
-# flake8: noqa: F401
 from sqlfluff.core.parser.segments.base import (
     BaseSegment,
     BaseFileSegment,
@@ -25,3 +24,25 @@
     Dedent,
     TemplateSegment,
 )
+
+__all__ = (
+    "BaseSegment",
+    "BaseFileSegment",
+    "UnparsableSegment",
+    "BracketedSegment",
+    "SegmentGenerator",
+    "RawSegment",
+    "CodeSegment",
+    "UnlexableSegment",
+    "CommentSegment",
+    "WhitespaceSegment",
+    "NewlineSegment",
+    "KeywordSegment",
+    "SymbolSegment",
+    "EphemeralSegment",
+    "allow_ephemeral",
+    "MetaSegment",
+    "Indent",
+    "Dedent",
+    "TemplateSegment",
+)

src/sqlfluff/core/templaters/__init__.py

@@ -1,7 +1,5 @@
 """Templater Code."""
 
-# flake8: noqa: F401
-
 from sqlfluff.core.templaters.base import TemplatedFile
 
 # Although these shouldn't usually be instantiated from here
@@ -15,3 +13,13 @@
 def core_templaters():
     """Returns the templater tuples for the core templaters."""
     yield from [RawTemplater, JinjaTemplater, PythonTemplater, PlaceholderTemplater]
+
+
+__all__ = (
+    "TemplatedFile",
+    "RawTemplater",
+    "JinjaTemplater",
+    "PythonTemplater",
+    "PlaceholderTemplater",
+    "core_templaters",
+)