Untitled
unknown
python
2 years ago
1.0 kB
7
Indexable
import psycopg2
import csv
db_params = {
"dbname": "",
"user": "",
"password": "",
"host": "",
"port": ""
}
output_file = "postgresdb_data.csv"
try:
connection = psycopg2.connect(**db_params)
cursor = connection.cursor()
query = "SELECT * FROM (table_name)"
cursor.execute(query)
rows = cursor.fetchall()
with open(output_file, "w", newline="") as csvfile:
csv_writer = csv.writer(csvfile)
# Write the header (column names) to the CSV file
column_names = [desc[0] for desc in cursor.description]
csv_writer.writerow(column_names)
# Write the data rows to the CSV file
csv_writer.writerows(rows)
print(f"Data saved to {output_file}")
except (Exception, psycopg2.Error) as error:
print("Error connecting to the database:", error)
finally:
if connection:
cursor.close()
connection.close()
print("Database connection closed.")Editor is loading...