hhNddlmZddlmZerddlmZdZGddeZdS)) TYPE_CHECKING)TreeSitterSegmenter)Languagez [ (create_table_statement) @create (select_statement) @select (insert_statement) @insert (update_statement) @update (delete_statement) @delete ] cdeZdZdZd dZdefdZdeeffd ZdefdZ dedefd Z xZ S) SQLSegmenteraCode segmenter for SQL. This class uses Tree-sitter to segment SQL code into its constituent statements (e.g., SELECT, CREATE TABLE). It also provides functionality to extract these statements and simplify the code into commented descriptions. returnrc$ddlm}|dS)z0Return the SQL language grammar for Tree-sitter.r) get_languagesql)tree_sitter_languagesr )selfr s v/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_community/document_loaders/parsers/language/sql.pyr zSQLSegmenter.get_languages$666666|E"""ctS)z2Return the Tree-sitter query for SQL segmentation.) CHUNK_QUERYr s rget_chunk_queryzSQLSegmenter.get_chunk_query#src\t}d|DS)zExtract SQL statements from the code. Ensures that all SQL statements end with a semicolon for consistency. cg|]T}|ds|dzn|US);)stripendswith.0stmts r z:SQLSegmenter.extract_functions_classes...s]   '+jjll&;&;C&@&@ RDJJLL3  djjll   r)superextract_functions_classes)r extracted __class__s rrz&SQLSegmenter.extract_functions_classes's; GG5577   !    rcddd|DS)zSimplify the extracted SQL code into comments. Converts SQL statements into commented descriptions for easy readability.  c<g|]}d|S)z -- Code for: )rrs rrz.SQLSegmenter.simplify_code..9s8   / ..   r)joinrrs r simplify_codezSQLSegmenter.simplify_code3sB yy   ::<<      rtextc d|S)z#Create a line comment in SQL style.z-- )r r&s rmake_line_commentzSQLSegmenter.make_line_comment?sT||r)rr) __name__ __module__ __qualname____doc__r strrlistrr%r) __classcell__)r s@rrrs####   49         s     ccrrN)typingrKlangchain_community.document_loaders.parsers.language.tree_sitter_segmenterr tree_sitterrrrr(rrr4s %$$$$$$ ,,,,,&,,,,,r