Merge pull request #104 from swiatek25/fix_tblproperties_parsing

fixed > 2 tblproperties parsing
xnuinside · Jan 4, 2022 · 925f570 · 925f570
2 parents b46f99b + e649a26
commit 925f570
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/simple_ddl_parser/dialects/hql.py b/simple_ddl_parser/dialects/hql.py
@@ -44,7 +44,7 @@ def p_expression_tblproperties(self, p):
  def p_multi_assigments(self, p):
  """multi_assigments : LP assigment
  | multi_assigments RP
- | multi_assigments COMMA assigment RP"""
+ | multi_assigments COMMA assigment"""
  p_list = remove_par(list(p))
  p[0] = p_list[1]
  p[0].update(p_list[-1])

diff --git a/tests/test_hql_output_mode.py b/tests/test_hql_output_mode.py
@@ -1718,7 +1718,9 @@ def test_table_properties():
  STORED AS PARQUET LOCATION 'hdfs:https://test'
  TBLPROPERTIES (
  'parquet.compression'='SNAPPY',
- 'parquet.compression2'='SNAPPY2'
+ 'parquet.compression2'='SNAPPY2',
+ 'parquet.compression3'='SNAPPY3',
+ 'parquet.compression4'='SNAPPY4'
  )
  """
  result = DDLParser(ddl).run(group_by_type=True, output_mode="hql")
@@ -1762,6 +1764,8 @@ def test_table_properties():
  "tblproperties": {
  "'parquet.compression'": "'SNAPPY'",
  "'parquet.compression2'": "'SNAPPY2'",
+ "'parquet.compression3'": "'SNAPPY3'",
+ "'parquet.compression4'": "'SNAPPY4'",
  },
  }
  ],