fix: update with join

Author: long2ice

pypika/dialects/postgresql.py

@@ -160,7 +160,7 @@ def get_sql(self, with_alias: bool = False, subquery: bool = False, **kwargs: An
             querystring += self._set_sql(**kwargs)
 
             if self._joins:
-                self._from.append(self._update_table)
+                self._from.append(self._update_table.as_(self._update_table.get_table_name() + "_"))
 
             if self._from:
                 querystring += self._from_sql(**kwargs)

pypika/dialects/sqlite.py

@@ -65,7 +65,7 @@ def get_sql(self, **kwargs: Any) -> str:
             querystring += self._set_sql(**kwargs)
 
             if self._joins:
-                self._from.append(self._update_table)
+                self._from.append(self._update_table.as_(self._update_table.get_table_name() + "_"))
 
             if self._from:
                 querystring += self._from_sql(**kwargs)

tests/dialects/test_postgresql.py

@@ -297,6 +297,6 @@ def test_return_field_from_join_table(self):
         )
 
         self.assertEqual(
-            'UPDATE "abc" SET "foo"=\'bar\' FROM "abc" JOIN "xyz" ON "xyz"."id"="abc"."xyz" WHERE "abc"."foo"=0 RETURNING "xyz"."a"',
+            'UPDATE "abc" SET "foo"=\'bar\' FROM "abc" "abc_" JOIN "xyz" ON "xyz"."id"="abc"."xyz" WHERE "abc"."foo"=0 RETURNING "xyz"."a"',
             str(q),
         )

tests/test_updates.py

@@ -176,7 +176,7 @@ def test_update_with_join(self):
             .set(self.table_abc.lname, self.table_def.lname)
         )
         self.assertEqual(
-            'UPDATE "abc" SET "lname"="def"."lname" FROM "abc" JOIN "def" ON "def"."abc_id"="abc"."id"',
+            'UPDATE "abc" SET "lname"="def"."lname" FROM "abc" "abc_" JOIN "def" ON "def"."abc_id"="abc"."id"',
             str(q),
         )
 
@@ -207,7 +207,7 @@ def test_update_with_join(self):
             .set(self.table_abc.lname, self.table_def.lname)
         )
         self.assertEqual(
-            'UPDATE "abc" SET "lname"="def"."lname" FROM "abc" JOIN "def" ON "def"."abc_id"="abc"."id"',
+            'UPDATE "abc" SET "lname"="def"."lname" FROM "abc" "abc_" JOIN "def" ON "def"."abc_id"="abc"."id"',
             str(q),
         )