add sql expression for repartition

Author: milenkovicm

python/datafusion/dataframe.py

@@ -914,17 +914,20 @@ def repartition(self, num: int) -> DataFrame:
         """
         return DataFrame(self.df.repartition(num))
 
-    def repartition_by_hash(self, *exprs: Expr, num: int) -> DataFrame:
+    def repartition_by_hash(self, *exprs: Expr | str, num: int) -> DataFrame:
         """Repartition a DataFrame using a hash partitioning scheme.
 
         Args:
-            exprs: Expressions to evaluate and perform hashing on.
+            exprs: Expressions or a SQL expression string to evaluate
+                   and perform hashing on.
             num: Number of partitions to repartition the DataFrame into.
 
         Returns:
             Repartitioned DataFrame.
         """
-        exprs = [expr.expr for expr in exprs]
+        exprs = [self.parse_sql_expr(e) if isinstance(e, str) else e for e in exprs]
+        exprs = expr_list_to_raw_expr_list(exprs)
+
         return DataFrame(self.df.repartition_by_hash(*exprs, num=num))
 
     def union(self, other: DataFrame, distinct: bool = False) -> DataFrame:

python/tests/test_dataframe.py

@@ -1613,6 +1613,11 @@ def test_repartition(df):
 def test_repartition_by_hash(df):
     df.repartition_by_hash(column("a"), num=2)
 
+def test_repartition_by_hash_sql_expression(df):
+    df.repartition_by_hash("a", num=2)
+
+def test_repartition_by_hash_mix(df):
+    df.repartition_by_hash(column("a"), "b", num=2)
 
 def test_intersect():
     ctx = SessionContext()