diff options
author | Amjith Ramanujam <amjith.r@gmail.com> | 2022-06-11 16:27:15 -0700 |
---|---|---|
committer | Amjith Ramanujam <amjith.r@gmail.com> | 2022-07-13 10:08:30 -0700 |
commit | 40f7a020dcef3db8e118e8eeee3c6a856885b85c (patch) | |
tree | de7b92c4c85106ed92f6627076c2743f96a0abab | |
parent | 687c5cf4d13197ad4c60747272186237d26db8fc (diff) |
Identify TiDB in the prompt.
-rw-r--r-- | mycli/sqlexecute.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/mycli/sqlexecute.py b/mycli/sqlexecute.py index f0cefc4..c019707 100644 --- a/mycli/sqlexecute.py +++ b/mycli/sqlexecute.py @@ -28,6 +28,7 @@ class ServerSpecies(enum.Enum): MySQL = 'MySQL' MariaDB = 'MariaDB' Percona = 'Percona' + TiDB = 'TiDB' Unknown = 'MySQL' @@ -55,6 +56,7 @@ class ServerInfo: re_species = ( (r'(?P<version>[0-9\.]+)-MariaDB', ServerSpecies.MariaDB), + (r'(?P<version>[0-9\.]+)[a-z0-9]*-TiDB', ServerSpecies.TiDB), (r'(?P<version>[0-9\.]+)[a-z0-9]*-(?P<comment>[0-9]+$)', ServerSpecies.Percona), (r'(?P<version>[0-9\.]+)[a-z0-9]*-(?P<comment>[A-Za-z0-9_]+)', |