PyPI - MindsDB - Versions diffs - 1.3.1__py3-none-any.whl → 24.12.4.0__py3-none-any.whl - Mend

MindsDB 1.3.1py3-none-any.whl → 24.12.4.0py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of MindsDB might be problematic. Click here for more details.

Files changed (1959) hide show

mindsdb/integrations/handlers/mssql_handler/connection_args.py ADDED Viewed

@@ -0,0 +1,52 @@
+from collections import OrderedDict
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+connection_args = OrderedDict(
+    user={
+        'type': ARG_TYPE.STR,
+        'description': 'The user name used to authenticate with the Microsoft SQL Server.',
+        'required': True,
+        'label': 'User'
+    },
+    password={
+        'type': ARG_TYPE.PWD,
+        'description': 'The password to authenticate the user with the Microsoft SQL Server.',
+        'required': True,
+        'label': 'Password',
+        'secret': True
+    },
+    database={
+        'type': ARG_TYPE.STR,
+        'description': 'The database name to use when connecting with the Microsoft SQL Server.',
+        'required': True,
+        'label': 'Database'
+    },
+    host={
+        'type': ARG_TYPE.STR,
+        'description': 'The host name or IP address of the Microsoft SQL Server.',
+        'required': True,
+        'label': 'Host'
+    },
+    port={
+        'type': ARG_TYPE.INT,
+        'description': 'The TCP/IP port of the Microsoft SQL Server. Must be an integer.',
+        'required': False,
+        'label': 'Port'
+    },
+    server={
+        'type': ARG_TYPE.STR,
+        'description': 'The server name of the Microsoft SQL Server. Typically only used with named instances or Azure SQL Database.',
+        'required': False,
+        'label': 'Server'
+    }
+)
+connection_args_example = OrderedDict(
+    host='127.0.0.1',
+    port=1433,
+    user='sa',
+    password='password',
+    database='master'
+)

mindsdb/integrations/handlers/mssql_handler/icon.svg ADDED Viewed

@@ -0,0 +1,21 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <path d="M16.8116 20.1133L24.0695 17.6432V17.6647L24.249 18.1096C23.3546 18.3766 23.2899 18.4011 23.2664 18.4165C23.2602 18.4957 23.2662 18.5754 23.284 18.6527C23.4298 19.5397 23.5212 20.4354 23.5576 21.3345C23.59 21.9635 23.5753 23.8229 23.537 24.3783C23.4327 26.0294 23.2124 27.6702 22.878 29.2877C22.2675 31.7437 21.4054 34.1242 20.3067 36.3879C20.0954 36.8686 19.8587 37.3368 19.5977 37.7901C19.2782 37.807 18.9581 37.807 18.6386 37.7901C16.1349 37.6328 13.6476 37.2634 11.2013 36.6855C10.0111 36.3656 8.84491 35.9554 7.71207 35.4582C5.97923 34.6635 4.4935 33.4944 3.74623 32.3499C3.46482 31.947 3.27938 31.48 3.2052 30.9876C3.13103 30.4951 3.17026 29.9913 3.31966 29.5178C3.57345 28.8367 3.93219 28.2035 4.38171 27.643C4.81708 27.1026 5.2986 26.6045 5.82035 26.1549C6.21163 25.8235 6.73237 25.3663 7.29135 24.8631C7.80325 24.4059 8.27397 23.9518 8.72409 23.5069L10.4893 23.019L16.8116 20.1133Z" fill="url(#paint0_linear_155_1187)"/>
+  <path d="M16.5881 0C15.4605 0.357648 14.3509 0.774457 13.2637 1.24883C11.9262 1.77674 10.6446 2.44715 9.43906 3.24941C9.15022 3.4586 8.90004 3.72056 8.70062 4.02264C8.67654 4.08395 8.66454 4.14969 8.66532 4.21595L9.40082 4.94315L11.166 5.52614L15.3289 6.30551L20.0891 7.15851L20.1391 6.72894H20.095L19.4713 6.62462L19.3447 6.39142C18.6975 5.20089 17.9826 3.72501 17.5795 2.73086C17.2642 1.98099 16.9969 1.21016 16.7793 0.423437C16.6852 0.0337536 16.6734 0.0122795 16.6146 0.00614273L16.5881 0ZM16.4998 0.291499C16.4998 0.291499 16.5263 0.457192 16.5469 0.647431C16.645 1.46719 16.8235 2.2743 17.0794 3.0561C17.2824 3.68512 17.2824 3.65137 17.0471 3.57466C16.4881 3.4151 13.9815 2.96098 12.1663 2.70017C11.8721 2.66028 11.6279 2.62039 11.6279 2.61426C11.6279 2.60812 12.94 1.86558 13.5255 1.58022C14.2786 1.21815 16.341 0.322183 16.494 0.291499H16.4998ZM11.2043 2.85665L11.4161 2.9303C13.2799 3.48233 15.1869 3.86122 17.1147 4.06253L17.456 4.09628C17.3455 4.16955 17.2294 4.23318 17.1088 4.28651C16.3321 4.69154 15.476 5.18862 14.8847 5.57216C14.7712 5.65181 14.6511 5.72062 14.5257 5.77775C14.5052 5.77775 14.3904 5.75627 14.2698 5.73786L14.0491 5.70411L13.5019 5.14259C12.537 4.16378 11.7779 3.40897 11.4867 3.12361L11.2043 2.85665ZM10.9836 3.03155L11.7574 4.03798C12.1781 4.59335 12.6076 5.13645 12.7017 5.26533C12.7609 5.33465 12.8149 5.4085 12.8635 5.48625C12.8429 5.50466 11.7456 5.28067 11.1631 5.14259C10.5806 5.00452 10.3187 4.93088 9.95097 4.81121L9.65677 4.70995V4.63018C9.65677 4.24663 10.1275 3.67285 10.9189 3.09599L10.9895 3.04383L10.9836 3.03155ZM17.5531 4.40619C17.5766 4.40619 17.6031 4.45834 17.6708 4.61176C17.8561 5.04133 18.4386 6.19504 18.5798 6.41904C18.624 6.49268 18.7005 6.49881 17.9267 6.36687C16.0674 6.06003 15.4701 5.94957 15.4701 5.9373C15.5102 5.90534 15.5524 5.87662 15.5966 5.85139C16.1897 5.50518 16.7502 5.10144 17.2706 4.64552L17.5325 4.41539C17.5325 4.41539 17.5501 4.40005 17.5531 4.41539V4.40619Z" fill="url(#paint1_linear_155_1187)"/>
+  <path d="M8.73847 3.94275C8.70231 4.02018 8.68352 4.10517 8.68352 4.19129C8.68352 4.27741 8.70231 4.3624 8.73847 4.43982C8.87079 4.68304 9.04538 4.89843 9.25332 5.07498C9.25332 5.07498 11.6951 7.55729 11.9893 7.91322C13.2889 9.39627 14.0007 11.3383 13.9811 13.3473C14.0159 14.7706 13.7191 16.1815 13.1162 17.4589C11.9747 20.0578 9.56515 22.9237 5.85236 26.1025L6.39662 25.9153C7.01879 25.4677 7.6695 25.0648 8.34421 24.7095C11.2615 23.0753 14.2878 21.6623 17.3997 20.4813C22.5335 18.4653 30.974 16.1088 35.7783 15.3417L36.2785 15.262L36.202 15.1362C35.8635 14.576 35.4952 14.0362 35.0987 13.5191C34.0224 12.1413 32.6995 10.9949 31.2035 10.1439C29.0559 8.89203 26.2727 7.91322 22.7512 7.18602C22.0892 7.04794 20.63 6.78406 19.4443 6.59689C16.9319 6.19186 15.3079 5.90957 13.5221 5.59046C12.8779 5.47693 11.9187 5.30511 11.2833 5.16089C10.7925 5.05202 10.3074 4.91683 9.82993 4.75587C9.43571 4.59324 8.86494 4.44904 8.74432 3.94889L8.73847 3.94275ZM10.1535 5.37261C10.2247 5.38968 10.2945 5.41224 10.3624 5.44011C10.5684 5.51376 10.839 5.59967 11.1568 5.69172L11.8805 5.89424C12.21 5.97708 12.4689 6.05992 12.4895 6.05992C12.5101 6.05992 13.0779 7.94698 13.2691 8.65577C13.3397 8.92579 13.3956 9.15284 13.3897 9.15284C13.3349 9.08444 13.2885 9.00919 13.2515 8.92885C12.5177 7.64208 11.5575 6.51151 10.4213 5.5966C10.3272 5.52841 10.2378 5.45357 10.1535 5.37261ZM12.8514 6.14891C12.9504 6.15989 13.0486 6.17732 13.1456 6.20107C14.0782 6.41893 15.7492 6.75031 16.8201 6.94055C16.9306 6.95048 17.0393 6.97523 17.1437 7.01419C17.0991 7.05507 17.0484 7.08822 16.9937 7.11238C16.8142 7.2075 16.0846 7.66161 15.8433 7.83344C15.2875 8.21182 14.7674 8.64436 14.29 9.12523L13.9958 9.43207C13.9711 9.36078 13.9505 9.28803 13.934 9.21421C13.6792 8.25337 13.3506 7.31551 12.9514 6.40972C12.8925 6.27778 12.8425 6.15811 12.8425 6.14891C12.8425 6.1397 12.8425 6.14891 12.8425 6.14891H12.8514ZM17.6027 7.09397C17.6973 7.29473 17.7636 7.50863 17.7998 7.72912C18.0049 8.58537 18.093 9.46754 18.0616 10.3495C18.0616 10.5888 18.041 10.8128 18.0293 10.8404L18.0116 10.8987L17.7174 10.8006C17.129 10.6072 16.1787 10.3158 15.3638 10.0734C14.896 9.94141 14.5195 9.82175 14.5195 9.81254C14.8244 9.45613 15.1486 9.11807 15.4903 8.79998C16.1629 8.19264 16.866 7.62316 17.5968 7.09397H17.6027ZM17.9792 7.15226C17.9969 7.13385 20.2505 7.53888 21.2772 7.74753C22.0392 7.90094 23.1454 8.14028 23.2131 8.17096C23.2807 8.20165 23.1307 8.25074 22.7629 8.42257C21.5114 8.99119 20.3096 9.67185 19.1708 10.4569C18.8766 10.6533 18.6559 10.8128 18.65 10.8128C18.6426 10.6912 18.6426 10.5693 18.65 10.4477C18.6548 9.37905 18.4547 8.32019 18.0616 7.3333C18.0275 7.27652 17.9998 7.21575 17.9792 7.15226ZM23.8103 8.35199C23.7919 8.60687 23.7515 8.85951 23.6897 9.10682C23.3643 10.2658 22.8998 11.3772 22.3069 12.4145C22.2382 12.5492 22.1615 12.6793 22.0774 12.8042C21.9628 12.7528 21.8525 12.6912 21.7479 12.6201C21.0557 12.2036 20.3372 11.8366 19.5973 11.5216C19.3796 11.4234 19.1855 11.3437 19.1796 11.3345C19.1737 11.3252 20.9095 10.0795 21.8274 9.49343C22.5658 9.02397 23.775 8.32745 23.7956 8.34893L23.8103 8.35199ZM24.228 8.42257C24.278 8.42257 25.2754 8.70793 25.7961 8.86749C27.0679 9.26315 28.3156 9.73859 29.5324 10.2912L29.9326 10.4814L29.6531 10.5489C27.4951 11.0505 25.3797 11.7339 23.3278 12.5925C23.1689 12.66 23.0336 12.7183 23.0159 12.7183C23.0438 12.617 23.0823 12.5193 23.1307 12.4268C23.6899 11.2289 24.0481 9.93996 24.1898 8.61588C24.1898 8.50849 24.2104 8.42257 24.228 8.42257ZM14.2871 10.8006C14.3047 10.7852 15.0667 10.9724 15.4639 11.0921C16.0905 11.2762 17.4173 11.7395 17.4173 11.7732C17.4173 11.807 17.2673 11.9113 17.0937 12.0801C16.3758 12.6937 15.6845 13.3626 14.8548 14.1972C14.6077 14.4427 14.3988 14.6422 14.39 14.6422C14.3812 14.6422 14.3724 14.6084 14.39 14.5624C14.512 13.4143 14.4863 12.2545 14.3135 11.1135C14.3135 10.9448 14.2841 10.8067 14.2871 10.8006ZM30.3091 10.819C30.3091 10.819 29.959 11.4326 29.7207 11.761C29.3912 12.255 28.9088 12.9055 27.8173 14.3353C27.2289 15.0901 26.5934 15.9431 26.3786 16.2285C26.1639 16.5139 25.9785 16.7563 25.9726 16.7563C25.9165 16.6834 25.8654 16.6065 25.8197 16.5261C25.2025 15.5659 24.4542 14.7048 23.5985 13.9702C23.4337 13.826 23.2543 13.6633 23.1925 13.6265C23.1527 13.5982 23.1162 13.5653 23.0836 13.5283C23.0836 13.513 24.0162 13.0957 24.7223 12.7981C25.9638 12.2703 27.6643 11.6413 28.9205 11.2363C29.5825 11.0184 30.2915 10.8067 30.3003 10.819H30.3091ZM30.7328 10.9325C30.843 10.9713 30.9478 11.025 31.0446 11.0921C32.3768 11.8873 33.6091 12.852 34.7133 13.964C35.0075 14.2709 35.7253 15.041 35.7136 15.0472L35.1634 15.0901C32.4314 15.345 29.7219 15.8158 27.0582 16.4985L26.6846 16.5844C26.6846 16.5844 26.8876 16.3604 27.1553 16.0935C28.8234 14.4151 29.5854 13.3596 30.4856 11.4695C30.6122 11.1902 30.721 10.9509 30.7328 10.9325ZM18.5176 12.2366C18.5941 12.255 19.3061 12.6017 19.8445 12.884C20.3829 13.1663 21.0801 13.5529 21.1184 13.5866C21.1566 13.6204 20.8595 13.737 20.53 13.8935C19.4973 14.4304 18.6147 14.9398 17.6939 15.5228C17.432 15.6885 17.2114 15.8296 17.2055 15.8296C17.1996 15.8296 17.2055 15.8051 17.3379 15.5228C17.8498 14.5326 18.2242 13.4716 18.45 12.3716C18.4735 12.2856 18.4941 12.2273 18.5118 12.2273L18.5176 12.2366ZM17.8057 12.3746C17.8204 12.39 17.635 13.0957 17.5115 13.4792C17.253 14.2814 16.9282 15.0586 16.5406 15.802C16.4523 15.9616 16.32 16.1948 16.2464 16.3267L16.1081 16.5568L15.8139 16.25C15.5155 15.9345 15.1709 15.6704 14.7931 15.4675C14.6994 15.4235 14.6106 15.369 14.5283 15.3049C14.5283 15.262 15.4668 14.3844 16.1876 13.7339C16.7024 13.2706 17.788 12.3593 17.8057 12.3777V12.3746ZM22.1657 14.2463L22.4334 14.4274C23.0906 14.8756 23.7222 15.3633 24.3251 15.8879C24.6399 16.1549 25.243 16.7225 25.3637 16.8606L25.4284 16.9342L24.9841 17.0662C22.4717 17.7903 20.53 18.4377 18.2646 19.3091C18.0116 19.4043 17.7998 19.484 17.7821 19.484C17.7645 19.484 17.7233 19.5147 18.2882 18.9686C19.6742 17.6516 20.9075 16.1697 21.9627 14.5532L22.1569 14.2463H22.1657ZM21.0183 14.5532C21.0183 14.5532 20.277 15.6701 19.8415 16.2929C19.3061 17.0386 18.3499 18.2843 17.6939 19.0974C17.4173 19.435 17.182 19.7111 17.1731 19.7111C17.1643 19.7111 17.1496 19.6313 17.1496 19.4994C17.1428 18.7857 16.9841 18.0824 16.6848 17.4405C16.5583 17.1766 16.5347 17.1153 16.5642 17.0846C16.6671 16.9895 18.2646 16.0383 19.2737 15.4706C19.9533 15.0932 21.0154 14.5501 21.0389 14.5501L21.0183 14.5532ZM14.084 16.3267C14.1853 16.3698 14.2836 16.42 14.3782 16.477C14.7126 16.6659 15.0289 16.8879 15.3226 17.1398C15.2285 17.2456 15.1217 17.3384 15.0049 17.416C14.4842 17.8026 13.6928 18.4285 13.2397 18.8121C12.7866 19.1956 12.7396 19.2232 12.7954 19.1373C13.0702 18.7151 13.316 18.2732 13.5309 17.8148C13.7056 17.4264 13.8589 17.0279 13.9899 16.6213C14.0108 16.5137 14.0475 16.4102 14.0988 16.3144L14.084 16.3267ZM15.9346 17.8394C15.961 17.8394 15.9934 17.8854 16.1434 18.1125C16.4617 18.6145 16.6706 19.1833 16.7554 19.7786V19.8921L16.0052 20.199C14.6607 20.7421 13.4221 21.279 12.5807 21.681C12.3453 21.7945 11.9335 21.9878 11.6657 22.1382C11.398 22.2885 11.1774 22.3898 11.1774 22.3837C11.1774 22.3775 11.348 22.2456 11.5569 22.0768C13.0848 20.9421 14.4688 19.61 15.6757 18.1125C15.7904 17.9683 15.8934 17.8363 15.9052 17.8333L15.9346 17.8394ZM15.0843 18.0542C14.767 18.487 14.422 18.8971 14.0517 19.2815C13.0303 20.3757 11.8865 21.3379 10.6449 22.1474L10.3183 22.3652C10.2801 22.3867 10.3183 22.3284 10.9067 21.6841C11.2686 21.276 11.5451 20.9354 11.857 20.5119C12.0017 20.2928 12.1879 20.107 12.4042 19.9658C13.2015 19.3368 15.0608 18.0204 15.0843 18.0419V18.0542Z" fill="url(#paint2_radial_155_1187)"/>
+  <path d="M22.3007 37.8821H22.4919V36.2804H23.0009V36.0779H21.7917V36.2804H22.3007V37.8821Z" fill="#231F1F"/>
+  <path d="M23.4653 36.6792C23.4728 36.5607 23.4728 36.4418 23.4653 36.3233C23.4771 36.3863 23.4949 36.448 23.5183 36.5074L24.1067 37.882H24.2008L24.7892 36.4951C24.8039 36.4522 24.8186 36.3969 24.8451 36.3233C24.8389 36.4378 24.8389 36.5525 24.8451 36.667V37.8943H25.051V36.0932H24.8039L24.2685 37.3359C24.2685 37.3788 24.2126 37.4678 24.1861 37.566C24.1662 37.4919 24.1396 37.42 24.1067 37.3512L23.5712 36.0932H23.2947V37.8974H23.4888V36.6915L23.4653 36.6792Z" fill="#231F1F"/>
+  <defs>
+    <linearGradient id="paint0_linear_155_1187" x1="6.54161" y1="22.8764" x2="9.25018" y2="21.3421" gradientUnits="userSpaceOnUse">
+      <stop stop-color="#909CA9"/>
+      <stop offset="1" stop-color="#EDEDEE"/>
+    </linearGradient>
+    <linearGradient id="paint1_linear_155_1187" x1="6.60098" y1="-7.92548" x2="8.7351" y2="-7.92548" gradientUnits="userSpaceOnUse">
+      <stop stop-color="#939FAB"/>
+      <stop offset="1" stop-color="#DCDEE1"/>
+    </linearGradient>
+    <radialGradient id="paint2_radial_155_1187" cx="0" cy="0" r="1" gradientUnits="userSpaceOnUse" gradientTransform="translate(3.14193 10.64) rotate(-169.263) scale(2.64333 5.69624)">
+      <stop stop-color="#EE352C"/>
+      <stop offset="1" stop-color="#A91D22"/>
+    </radialGradient>
+  </defs>
+</svg>

mindsdb/integrations/handlers/mssql_handler/mssql_handler.py ADDED Viewed

@@ -0,0 +1,218 @@
+import pymssql
+from pymssql import OperationalError
+import pandas as pd
+from mindsdb_sql_parser import parse_sql
+from mindsdb_sql_parser.ast.base import ASTNode
+from mindsdb.integrations.libs.base import DatabaseHandler
+from mindsdb.utilities import log
+from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
+from mindsdb.integrations.libs.response import (
+    HandlerStatusResponse as StatusResponse,
+    HandlerResponse as Response,
+    RESPONSE_TYPE
+)
+logger = log.getLogger(__name__)
+class SqlServerHandler(DatabaseHandler):
+    """
+    This handler handles connection and execution of the Microsoft SQL Server statements.
+    """
+    name = 'mssql'
+    def __init__(self, name, **kwargs):
+        super().__init__(name)
+        self.parser = parse_sql
+        self.connection_args = kwargs.get('connection_data')
+        self.dialect = 'mssql'
+        self.database = self.connection_args.get('database')
+        self.renderer = SqlalchemyRender('mssql')
+        self.connection = None
+        self.is_connected = False
+    def __del__(self):
+        if self.is_connected is True:
+            self.disconnect()
+    def connect(self):
+        """
+        Establishes a connection to a Microsoft SQL Server database.
+        Raises:
+            pymssql._mssql.OperationalError: If an error occurs while connecting to the Microsoft SQL Server database.
+        Returns:
+            pymssql.Connection: A connection object to the Microsoft SQL Server database.
+        """
+        if self.is_connected is True:
+            return self.connection
+        # Mandatory connection parameters
+        if not all(key in self.connection_args for key in ['host', 'user', 'password', 'database']):
+            raise ValueError('Required parameters (host, user, password, database) must be provided.')
+        config = {
+            'host': self.connection_args.get('host'),
+            'user': self.connection_args.get('user'),
+            'password': self.connection_args.get('password'),
+            'database': self.connection_args.get('database')
+        }
+        # Optional connection parameters
+        if 'port' in self.connection_args:
+            config['port'] = self.connection_args.get('port')
+        if 'server' in self.connection_args:
+            config['server'] = self.connection_args.get('server')
+        try:
+            self.connection = pymssql.connect(**config)
+            self.is_connected = True
+            return self.connection
+        except OperationalError as e:
+            logger.error(f'Error connecting to Microsoft SQL Server {self.database}, {e}!')
+            self.is_connected = False
+            raise
+    def disconnect(self):
+        """
+        Closes the connection to the Microsoft SQL Server database if it's currently open.
+        """
+        if not self.is_connected:
+            return
+        self.connection.close()
+        self.is_connected = False
+    def check_connection(self) -> StatusResponse:
+        """
+        Checks the status of the connection to the Microsoft SQL Server database.
+        Returns:
+            StatusResponse: An object containing the success status and an error message if an error occurs.
+        """
+        response = StatusResponse(False)
+        need_to_close = self.is_connected is False
+        try:
+            connection = self.connect()
+            with connection.cursor() as cur:
+                # Execute a simple query to test the connection
+                cur.execute('select 1;')
+            response.success = True
+        except OperationalError as e:
+            logger.error(f'Error connecting to Microsoft SQL Server {self.database}, {e}!')
+            response.error_message = str(e)
+        if response.success and need_to_close:
+            self.disconnect()
+        elif not response.success and self.is_connected:
+            self.is_connected = False
+        return response
+    def native_query(self, query: str) -> Response:
+        """
+        Executes a SQL query on the Microsoft SQL Server database and returns the result.
+        Args:
+            query (str): The SQL query to be executed.
+        Returns:
+            Response: A response object containing the result of the query or an error message.
+        """
+        need_to_close = self.is_connected is False
+        connection = self.connect()
+        with connection.cursor(as_dict=True) as cur:
+            try:
+                cur.execute(query)
+                if cur.description:
+                    result = cur.fetchall()
+                    response = Response(
+                        RESPONSE_TYPE.TABLE,
+                        data_frame=pd.DataFrame(
+                            result,
+                            columns=[x[0] for x in cur.description]
+                        )
+                    )
+                else:
+                    response = Response(RESPONSE_TYPE.OK)
+                connection.commit()
+            except Exception as e:
+                logger.error(f'Error running query: {query} on {self.database}, {e}!')
+                response = Response(
+                    RESPONSE_TYPE.ERROR,
+                    error_code=0,
+                    error_message=str(e)
+                )
+                connection.rollback()
+        if need_to_close is True:
+            self.disconnect()
+        return response
+    def query(self, query: ASTNode) -> Response:
+        """
+        Executes a SQL query represented by an ASTNode and retrieves the data.
+        Args:
+            query (ASTNode): An ASTNode representing the SQL query to be executed.
+        Returns:
+            Response: The response from the `native_query` method, containing the result of the SQL query execution.
+        """
+        query_str = self.renderer.get_string(query, with_failback=True)
+        logger.debug(f"Executing SQL query: {query_str}")
+        return self.native_query(query_str)
+    def get_tables(self) -> Response:
+        """
+        Retrieves a list of all non-system tables and views in the current schema of the Microsoft SQL Server database.
+        Returns:
+            Response: A response object containing the list of tables and views, formatted as per the `Response` class.
+        """
+        query = f"""
+            SELECT
+                table_schema,
+                table_name,
+                table_type
+            FROM {self.database}.INFORMATION_SCHEMA.TABLES
+            WHERE TABLE_TYPE in ('BASE TABLE', 'VIEW');
+        """
+        return self.native_query(query)
+    def get_columns(self, table_name) -> Response:
+        """
+        Retrieves column details for a specified table in the Microsoft SQL Server database.
+        Args:
+            table_name (str): The name of the table for which to retrieve column information.
+        Returns:
+            Response: A response object containing the column details, formatted as per the `Response` class.
+        Raises:
+            ValueError: If the 'table_name' is not a valid string.
+        """
+        query = f"""
+            SELECT
+                column_name as "Field",
+                data_type as "Type"
+            FROM
+                information_schema.columns
+            WHERE
+                table_name = '{table_name}'
+        """
+        return self.native_query(query)

mindsdb/integrations/handlers/mssql_handler/requirements.txt ADDED Viewed

	@@ -0,0 +1 @@
1	+ pymssql >= 2.1.4

mindsdb/integrations/handlers/mssql_handler/tests/__init__.py ADDED Viewed

File without changes

mindsdb/integrations/handlers/mssql_handler/tests/test_mssql_handler.py ADDED Viewed

@@ -0,0 +1,169 @@
+import os
+import pytest
+import pymssql
+import pandas as pd
+from mindsdb.integrations.handlers.mssql_handler.mssql_handler import SqlServerHandler
+from mindsdb.integrations.libs.response import (
+    HandlerResponse as Response,
+    RESPONSE_TYPE,
+)
+HANDLER_KWARGS = {
+    "connection_data": {
+        "host": os.environ.get("MDB_TEST_MSSQL_HOST", "localhost"),
+        "port": os.environ.get("MDB_TEST_MSSQL_PORT", "1433"),
+        "user": os.environ.get("MDB_TEST_MSSQL_USER", "sa"),
+        "password": os.environ.get("MDB_TEST_MSSQL_PASSWORD", "admin5678@"),
+        "database": os.environ.get("MDB_TEST_MSSQL_DATABASE", "mdb_db_handler_test"),
+    }
+}
+@pytest.fixture(scope="class")
+def sql_server_handler():
+    seed_db()
+    handler = SqlServerHandler("test_sqlserver_handler", **HANDLER_KWARGS)
+    yield handler
+    handler.disconnect()
+def seed_db():
+    """Seed the test DB with some data"""
+    # Connect to 'master' while we create our test DB
+    conn_info = HANDLER_KWARGS["connection_data"].copy()
+    conn_info["database"] = "master"
+    db = pymssql.connect(**conn_info, autocommit=True)
+    cursor = db.cursor()
+    with open("mindsdb/integrations/handlers/mssql_handler/tests/seed.sql", "r") as f:
+        for line in f.readlines():
+            cursor.execute(line)
+    cursor.close()
+    db.close()
+def check_valid_response(res):
+    if res.resp_type == RESPONSE_TYPE.TABLE:
+        assert res.data_frame is not None, "expected to have some data, but got None"
+    assert (
+        res.error_code == 0
+    ), f"expected to have zero error_code, but got {res.error_code}"
+    assert (
+        res.error_message is None
+    ), f"expected to have None in error message, but got {res.error_message}"
+def get_table_names(sql_server_handler):
+    res = sql_server_handler.get_tables()
+    tables = res.data_frame
+    assert tables is not None, "expected to have some tables in the db, but got None"
+    assert (
+        "table_name" in tables
+    ), f"expected to get 'table_name' column in the response:\n{tables}"
+    return list(tables["table_name"])
+@pytest.mark.usefixtures("sql_server_handler")
+class TestMSSQLHandlerConnect:
+    def test_connect(self, sql_server_handler):
+        sql_server_handler.connect()
+        assert sql_server_handler.is_connected, "the handler has failed to connect"
+    def test_check_connection(self, sql_server_handler):
+        res = sql_server_handler.check_connection()
+        assert res.success, res.error_message
+class TestMSSQLHandlerDisconnect:
+    def test_disconnect(self, sql_server_handler):
+        sql_server_handler.disconnect()
+        assert sql_server_handler.is_connected is False, "failed to disconnect"
+    def test_check_connection(self, sql_server_handler):
+        res = sql_server_handler.check_connection()
+        assert res.success, res.error_message
+@pytest.mark.usefixtures("sql_server_handler")
+class TestMSSQLHandlerTables:
+    table_for_creation = "test_mdb"
+    def test_get_tables(self, sql_server_handler):
+        res = sql_server_handler.get_tables()
+        tables = res.data_frame
+        assert (
+            tables is not None
+        ), "expected to have some tables in the db, but got None"
+        assert (
+            "table_name" in tables
+        ), f"expected to get 'table_name' in the response but got: {tables}"
+        assert (
+            "test" in tables["table_name"].values
+        ), "expected to have 'test' in the response."
+    def test_get_columns(self, sql_server_handler):
+        response = sql_server_handler.get_columns("test")
+        assert response.type == RESPONSE_TYPE.TABLE, "expected a TABLE"
+        assert len(response.data_frame) > 0, "expected > O columns"
+        expected_columns = {
+            "Field": ["col_one", "col_two", "col_three", "col_four"],
+            "Type": ["int", "int", "float", "varchar"],
+        }
+        expected_df = pd.DataFrame(expected_columns)
+        assert response.data_frame.equals(
+            expected_df
+        ), "response does not contain the expected columns"
+    def test_create_table(self, sql_server_handler):
+        query = f"""
+        IF NOT EXISTS (SELECT * FROM sys.tables WHERE name = '{self.table_for_creation}')
+        BEGIN
+            CREATE TABLE {self.table_for_creation} (test_col INT)
+        END
+        """
+        res = sql_server_handler.native_query(query)
+        check_valid_response(res)
+        tables = get_table_names(sql_server_handler)
+        assert (
+            self.table_for_creation in tables
+        ), f"expected to have {self.table_for_creation} in database, but got: {tables}"
+    def test_drop_table(self, sql_server_handler):
+        query = f"DROP TABLE IF EXISTS {self.table_for_creation}"
+        res = sql_server_handler.native_query(query)
+        check_valid_response(res)
+        tables = get_table_names(sql_server_handler)
+        assert self.table_for_creation not in tables
+@pytest.mark.usefixtures("sql_server_handler")
+class TestMSSQLHandlerQuery:
+    def test_select_native_query(self, sql_server_handler):
+        query = "SELECT * FROM test"
+        response = sql_server_handler.native_query(query)
+        assert type(response) is Response
+        assert response.resp_type == RESPONSE_TYPE.TABLE
+        expected_data = {
+            "col_one": [1, 2, 3],
+            "col_two": [-1, -2, -3],
+            "col_three": [0.1, 0.2, 0.3],
+            "col_four": ["A", "B", "C"],
+        }
+        expected_df = pd.DataFrame(expected_data)
+        assert response.data_frame.equals(
+            expected_df
+        ), "response does not contain the expected data"
+    def test_select_query(self, sql_server_handler):
+        limit = 3
+        query = "SELECT * FROM test"
+        res = sql_server_handler.query(query)
+        check_valid_response(res)
+        got_rows = res.data_frame.shape[0]
+        want_rows = limit
+        assert (
+            got_rows == want_rows
+        ), f"expected to have {want_rows} rows in response but got: {got_rows}"

mindsdb/integrations/handlers/mysql_handler/__about__.py ADDED Viewed

@@ -0,0 +1,9 @@
+__title__ = 'MindsDB MySQL handler'
+__package_name__ = 'mindsdb_mysql_handler'
+__version__ = '0.0.1'
+__description__ = "MindsDB handler for MySQL"
+__author__ = 'MindsDB Inc'
+__github__ = 'https://github.com/mindsdb/mindsdb'
+__pypi__ = 'https://pypi.org/project/mindsdb/'
+__license__ = 'MIT'
+__copyright__ = 'Copyright 2022- mindsdb'

mindsdb/integrations/handlers/mysql_handler/__init__.py ADDED Viewed

@@ -0,0 +1,20 @@
+from mindsdb.integrations.libs.const import HANDLER_TYPE
+from .__about__ import __version__ as version, __description__ as description
+from .connection_args import connection_args, connection_args_example
+try:
+    from .mysql_handler import MySQLHandler as Handler
+    import_error = None
+except Exception as e:
+    Handler = None
+    import_error = e
+title = 'MySQL'
+name = 'mysql'
+type = HANDLER_TYPE.DATA
+icon_path = 'icon.svg'
+__all__ = [
+    'Handler', 'version', 'name', 'type', 'title', 'description',
+    'connection_args', 'connection_args_example', 'import_error', 'icon_path'
+]

mindsdb/integrations/handlers/mysql_handler/connection_args.py ADDED Viewed

@@ -0,0 +1,76 @@
+from collections import OrderedDict
+from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE
+connection_args = OrderedDict(
+    url={
+        'type': ARG_TYPE.STR,
+        'description': 'The URI-Like connection string to the MySQL server. If provided, it will override the other connection arguments.',
+        'required': False,
+        'label': 'URL'
+    },
+    user={
+        'type': ARG_TYPE.STR,
+        'description': 'The user name used to authenticate with the MySQL server.',
+        'required': True,
+        'label': 'User'
+    },
+    password={
+        'type': ARG_TYPE.PWD,
+        'description': 'The password to authenticate the user with the MySQL server.',
+        'required': True,
+        'label': 'Password',
+        'secret': True
+    },
+    database={
+        'type': ARG_TYPE.STR,
+        'description': 'The database name to use when connecting with the MySQL server.',
+        'required': True,
+        'label': 'Database'
+    },
+    host={
+        'type': ARG_TYPE.STR,
+        'description': 'The host name or IP address of the MySQL server. NOTE: use \'127.0.0.1\' instead of \'localhost\' to connect to local server.',
+        'required': True,
+        'label': 'Host'
+    },
+    port={
+        'type': ARG_TYPE.INT,
+        'description': 'The TCP/IP port of the MySQL server. Must be an integer.',
+        'required': True,
+        'label': 'Port'
+    },
+    ssl={
+        'type': ARG_TYPE.BOOL,
+        'description': 'Set it to True to enable ssl.',
+        'required': False,
+        'label': 'ssl'
+    },
+    ssl_ca={
+        'type': ARG_TYPE.PATH,
+        'description': 'Path or URL of the Certificate Authority (CA) certificate file',
+        'required': False,
+        'label': 'ssl_ca'
+    },
+    ssl_cert={
+        'type': ARG_TYPE.PATH,
+        'description': 'Path name or URL of the server public key certificate file',
+        'required': False,
+        'label': 'ssl_cert'
+    },
+    ssl_key={
+        'type': ARG_TYPE.PATH,
+        'description': 'The path name or URL of the server private key file',
+        'required': False,
+        'label': 'ssl_key',
+    }
+)
+connection_args_example = OrderedDict(
+    host='127.0.0.1',
+    port=3306,
+    user='root',
+    password='password',
+    database='database'
+)

mindsdb/integrations/handlers/mysql_handler/icon.svg ADDED Viewed

@@ -0,0 +1,4 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="40" height="40" viewBox="0 0 40 40" fill="none">
+  <path fill-rule="evenodd" clip-rule="evenodd" d="M35.9307 30.1371C33.8099 30.0865 32.1655 30.3013 30.7862 30.8953C30.372 31.0596 29.7548 31.0596 29.7009 31.5735C29.9122 31.7883 29.937 32.138 30.1151 32.437C30.4775 33.0645 30.9442 33.6233 31.4945 34.0883C32.0537 34.5096 32.6087 34.9519 33.1927 35.3521C34.2241 36.0008 35.3922 36.3799 36.3987 37.037C36.9869 37.4161 37.5668 37.8795 38.1509 38.3007C38.4408 38.5156 38.6272 38.8694 39 39V38.92C38.8136 38.6756 38.7598 38.326 38.5858 38.0522C38.3248 37.7868 38.0556 37.5383 37.7905 37.2729C37.016 36.2231 36.0869 35.3009 35.036 34.5391C34.2076 33.9451 32.327 33.1321 31.9832 32.1338L31.9294 32.079C32.5507 31.994 33.1634 31.853 33.7602 31.6577C34.6632 31.4134 35.4833 31.4682 36.4112 31.2365C36.8254 31.127 37.2396 30.9922 37.6869 30.8574V30.6131C37.2065 30.1286 36.8585 29.4757 36.3615 29.0166C34.989 27.7949 33.5252 26.6837 31.9832 25.693C31.1548 25.1495 30.1027 24.7999 29.2246 24.3408C28.9098 24.1765 28.3962 24.0964 28.1932 23.8269C27.7572 23.1738 27.392 22.4746 27.1038 21.7417C26.3375 20.2589 25.5919 18.6077 24.9292 17.0364C24.4529 15.9833 24.1629 14.9302 23.5789 13.9571C20.8451 9.35718 17.8752 6.57698 13.3147 3.84733C12.333 3.27865 11.1691 3.03433 9.92228 2.73946C9.25954 2.70997 8.5968 2.65942 7.93406 2.62993C7.51984 2.44037 7.08492 1.92645 6.69142 1.68213C5.17954 0.709063 1.2818 -1.39714 0.167569 1.38727C-0.549018 3.14385 1.22795 4.87516 1.82442 5.76819C2.33675 6.39846 2.7889 7.07692 3.17475 7.79436C3.36115 8.25352 3.41499 8.74216 3.58896 9.22659C3.97562 10.463 4.43626 11.6742 4.96829 12.8535C5.24971 13.4406 5.57659 14.004 5.94583 14.5385C6.16122 14.8375 6.52988 14.9597 6.60858 15.461C6.23993 16.0002 6.19436 16.8132 5.99968 17.4872C5.04699 20.5454 5.41564 24.3281 6.77012 26.5733C7.18433 27.2515 8.19915 28.7385 9.55363 28.1699C10.7466 27.6812 10.4815 26.1395 10.8253 24.7999C10.904 24.4714 10.8501 24.2565 11.0117 24.0417V24.0248C11.3803 24.7789 11.7531 25.5076 12.0969 26.2658C13.0438 27.6578 14.2095 28.8817 15.5473 29.8885C16.181 30.3772 16.6864 31.2154 17.4817 31.5145V31.4303H17.4278C17.2497 31.2234 17.0449 31.042 16.8189 30.8911C16.3179 30.3882 15.8567 29.8459 15.4396 29.2693C14.317 27.716 13.3236 26.0702 12.4697 24.3492C12.0555 23.5067 11.6744 22.5884 11.3306 21.7501C11.1691 21.4258 11.1691 20.9413 10.9164 20.7771C10.5022 21.371 9.93885 21.8849 9.64476 22.6137C9.13942 23.7805 9.08971 25.2127 8.90331 26.6997C8.79562 26.7292 8.84947 26.6997 8.79562 26.7545C7.96719 26.5354 7.65653 25.6466 7.33759 24.8884C6.5423 22.9675 6.40976 19.884 7.10149 17.6683C7.28374 17.0996 8.08317 15.3178 7.76423 14.7744C7.60268 14.2604 7.07249 13.9656 6.78254 13.557C6.41185 13.0111 6.09155 12.4315 5.82571 11.8257C5.19197 10.3134 4.86888 8.63685 4.16886 7.12459C3.78935 6.39056 3.34574 5.6927 2.84339 5.03944C2.33805 4.31069 1.783 3.79677 1.38536 2.93322C1.25281 2.63414 1.06641 2.1455 1.27766 1.82535C1.29209 1.73254 1.33662 1.6473 1.40416 1.58319C1.47171 1.51907 1.55841 1.47976 1.65045 1.47151C1.99425 1.17243 2.97593 1.55155 3.3073 1.71584C4.22816 2.06697 5.10871 2.51895 5.93341 3.06381C6.3062 3.33762 6.70384 3.85153 7.17605 3.98633H7.73523C8.56366 4.17589 9.53706 4.0411 10.3323 4.28542C11.6765 4.74225 12.9594 5.3677 14.1514 6.14731C17.6703 8.4367 20.529 11.6346 22.4356 15.4146C22.7546 16.0339 22.8871 16.6025 23.1812 17.2513C23.7363 18.5782 24.4239 19.9304 24.983 21.2278C25.4691 22.5065 26.1017 23.7224 26.8677 24.8505C27.2612 25.4191 28.8559 25.714 29.5725 26.0131C30.194 26.2334 30.8037 26.4864 31.3992 26.7713C32.3022 27.3358 33.201 27.9845 34.0502 28.608C34.4644 28.9323 35.8023 29.6105 35.881 30.1497" fill="#00758F"/>
+  <path fill-rule="evenodd" clip-rule="evenodd" d="M9.02147 7.00005C8.67688 6.99818 8.33354 7.0521 8 7.16049V7.2237H8.05049C8.28967 7.67455 8.55583 8.10203 8.84669 8.50245C9.04477 8.98866 9.23508 9.4992 9.41763 10L9.46812 9.93679C9.65328 9.7528 9.7984 9.51333 9.89005 9.24053C9.9817 8.96772 10.0169 8.67039 9.99245 8.37603C9.88259 8.16475 9.78273 7.94558 9.69339 7.71965C9.5458 7.40847 9.21955 7.23343 9.02147 7.00005" fill="#00758F"/>
+</svg>

MindsDB 1.3.1__py3-none-any.whl → 24.12.4.0__py3-none-any.whl

Potentially problematic release.

MindsDB 1.3.1py3-none-any.whl → 24.12.4.0py3-none-any.whl