Pyspark Case When Postgresql