add _prepare_sorting() method, refactor order_by() method

Author: co0lc0der

simple_query_builder/querybuilder.py

@@ -336,18 +336,33 @@ def offset(self, offset: int = 0):
         self._sql += f" OFFSET {offset}"
         return self
 
-    def order_by(self, field: str = '', sort: str = 'ASC'):
-        if field == '' or sort == '':
-            self.set_error(f"Empty field or sort in {inspect.stack()[0][3]} method")
-            return self
+    def _prepare_sorting(self, field: str = '', sort: str = ''):
+        if field.find(' ') > -1:
+            splitted = field.split(' ')
+            field = splitted[0]
+            sort = splitted[1]
 
-        sort = sort.upper()
         field = field.replace('.', '`.`')
 
+        if sort == '':
+            sort = 'ASC'
+        else:
+            sort = sort.upper()
+
+        return field, sort
+
+    def order_by(self, field: str = '', sort: str = ''):
+        if field == '':
+            self.set_error(f"Empty field in {inspect.stack()[0][3]} method")
+            return self
+
+        field, sort = self._prepare_sorting(field, sort)
+
         if sort in self._SORT_TYPES:
             self._sql += f" ORDER BY `{field}` {sort}"
         else:
             self._sql += f" ORDER BY `{field}`"
+
         return self
 
     def group_by(self, field: str = ''):