summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAmjith Ramanujam <amjith.r@gmail.com>2022-06-11 16:27:15 -0700
committerAmjith Ramanujam <amjith.r@gmail.com>2022-07-13 10:08:30 -0700
commit40f7a020dcef3db8e118e8eeee3c6a856885b85c (patch)
treede7b92c4c85106ed92f6627076c2743f96a0abab
parent687c5cf4d13197ad4c60747272186237d26db8fc (diff)
Identify TiDB in the prompt.
-rw-r--r--mycli/sqlexecute.py2
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_]+)',