Import python venv for stability

2026-02-15 21:24:16 -08:00
parent 1343e93a59
commit 7d784705c9
4997 changed files with 1628270 additions and 0 deletions
@@ -0,0 +1,117 @@
+import json
+
+try:
+    import mysql.connector as mysql_connector
+except ImportError:
+    mysql_connector = None
+try:
+    import mariadb
+except ImportError:
+    mariadb = None
+
+from peewee import ImproperlyConfigured
+from peewee import Insert
+from peewee import MySQLDatabase
+from peewee import Node
+from peewee import NodeList
+from peewee import SQL
+from peewee import TextField
+from peewee import fn
+from peewee import __deprecated__
+
+
+class MySQLConnectorDatabase(MySQLDatabase):
+    def _connect(self):
+        if mysql_connector is None:
+            raise ImproperlyConfigured('MySQL connector not installed!')
+        return mysql_connector.connect(db=self.database, autocommit=True,
+                                       **self.connect_params)
+
+    def cursor(self, commit=None, named_cursor=None):
+        if commit is not None:
+            __deprecated__('"commit" has been deprecated and is a no-op.')
+        if self.is_closed():
+            if self.autoconnect:
+                self.connect()
+            else:
+                raise InterfaceError('Error, database connection not opened.')
+        return self._state.conn.cursor(buffered=True)
+
+    def get_binary_type(self):
+        return mysql_connector.Binary
+
+
+class MariaDBConnectorDatabase(MySQLDatabase):
+    def _connect(self):
+        if mariadb is None:
+            raise ImproperlyConfigured('mariadb connector not installed!')
+        self.connect_params.pop('charset', None)
+        self.connect_params.pop('sql_mode', None)
+        self.connect_params.pop('use_unicode', None)
+        return mariadb.connect(db=self.database, autocommit=True,
+                               **self.connect_params)
+
+    def cursor(self, commit=None, named_cursor=None):
+        if commit is not None:
+            __deprecated__('"commit" has been deprecated and is a no-op.')
+        if self.is_closed():
+            if self.autoconnect:
+                self.connect()
+            else:
+                raise InterfaceError('Error, database connection not opened.')
+        return self._state.conn.cursor(buffered=True)
+
+    def _set_server_version(self, conn):
+        version = conn.server_version
+        version, point = divmod(version, 100)
+        version, minor = divmod(version, 100)
+        self.server_version = (version, minor, point)
+        if self.server_version >= (10, 5, 0):
+            self.returning_clause = True
+
+    def last_insert_id(self, cursor, query_type=None):
+        if not self.returning_clause:
+            return cursor.lastrowid
+        elif query_type == Insert.SIMPLE:
+            try:
+                return cursor[0][0]
+            except (AttributeError, IndexError):
+                return cursor.lastrowid
+        return cursor
+
+    def get_binary_type(self):
+        return mariadb.Binary
+
+
+class JSONField(TextField):
+    field_type = 'JSON'
+
+    def __init__(self, json_dumps=None, json_loads=None, **kwargs):
+        self._json_dumps = json_dumps or json.dumps
+        self._json_loads = json_loads or json.loads
+        super(JSONField, self).__init__(**kwargs)
+
+    def python_value(self, value):
+        if value is not None:
+            try:
+                return self._json_loads(value)
+            except (TypeError, ValueError):
+                return value
+
+    def db_value(self, value):
+        if value is not None:
+            if not isinstance(value, Node):
+                value = self._json_dumps(value)
+            return value
+
+    def extract(self, path):
+        return fn.json_extract(self, path)
+
+
+def Match(columns, expr, modifier=None):
+    if isinstance(columns, (list, tuple)):
+        match = fn.MATCH(*columns)  # Tuple of one or more columns / fields.
+    else:
+        match = fn.MATCH(columns)  # Single column / field.
+    args = expr if modifier is None else NodeList((expr, SQL(modifier)))
+    return NodeList((match, fn.AGAINST(args)))