Merge pull request #59 from rlsalcido24/yeardateformat

year date format
techvaquero · Apr 2, 2024 · c9c0a1b · c9c0a1b
2 parents 185619f + 60d6890
commit c9c0a1b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/helper/_resources/config/redshift/syntax_mappings.json b/helper/_resources/config/redshift/syntax_mappings.json
@@ -73,8 +73,8 @@
 "customnoposix" : {"source_pattern": "!~",
                       "target_pattern": "not rlike"
                        },
-"tocharlower" : {"source_pattern": "to_char\\([^)]*?\\)",
-                      "target_pattern": "to_char(#arg0, lower(#arg1))"
+"yearformat" : {"source_pattern": "YYYY-",
+                      "target_pattern": "yyyy-"
                        },                                                                                                                
 "getdate_to_df" : {"source_pattern": "getdate\\(\\)",
                   "target_pattern": "date_format(date_trunc('second', current_timestamp()), 'yyyy-MM-dd HH:mm:ss')"

diff --git a/helper/convert_to_databricks.py b/helper/convert_to_databricks.py
@@ -598,7 +598,12 @@ def convert_syntax_expressions(content: str, source_pattern: str, target_pattern
   elif target_pattern == "not rlike":
     encontrar = re.findall(source_pattern, content, flags= re.DOTALL | re.IGNORECASE)
     num_matches = len(encontrar)
-    updated_content = re.sub(source_pattern, target_pattern, content, flags= re.DOTALL | re.IGNORECASE)              
+    updated_content = re.sub(source_pattern, target_pattern, content, flags= re.DOTALL | re.IGNORECASE)
+
+  elif target_pattern == "yyyy-":
+    encontrar = re.findall(source_pattern, content)
+    num_matches = len(encontrar)
+    updated_content = re.sub(source_pattern, target_pattern, content)                 
 
   else:
       initargs = findargs(content, source_pattern)