DROP/ TRUNCATE
を試しました シナリオですが、Glueで既に作成されている接続では実行できませんが、純粋なPythonPostgreSQLドライバー
- pg8000のtarをpypiからダウンロード
- 空の
__init__.py
を作成します ルートフォルダ内 - コンテンツを圧縮してS3にアップロード
-
Python lib path
でzipファイルを参照します 仕事の - DB接続の詳細をジョブパラメータとして設定します(すべてのキー名の前に必ず
--
を付けてください )。 [サーバー側の暗号化]チェックボックスをオンにします。
次に、接続を作成してSQLを実行するだけです。
import sys
from awsglue.utils import getResolvedOptions
from pyspark.context import SparkContext
from awsglue.context import GlueContext
from awsglue.dynamicframe import DynamicFrame
from awsglue.job import Job
import pg8000
args = getResolvedOptions(sys.argv, [
'JOB_NAME',
'PW',
'HOST',
'USER',
'DB'
])
# ...
# Create Spark & Glue context
job = Job(glueContext)
job.init(args['JOB_NAME'], args)
# ...
config_port = 5432
conn = pg8000.connect(
database=args['DB'],
user=args['USER'],
password=args['PW'],
host=args['HOST'],
port=config_port
)
query = "TRUNCATE TABLE {0};".format(".".join([schema, table]))
cur = conn.cursor()
cur.execute(query)
conn.commit()
cur.close()
conn.close()