Skip to main content

Interactive PostgreSQL Script

Do you want an input from end user, while running PostgreSQL script.

No Problem, here is the one of the solution for you.

In the following example, i am taking the confirmation from an end user, before dropping an existing database called "sample".


-- When any exception raised from this script, the complete script execution is going to fail.

-- We will be raising a custom exception, when the Drop DB != 'y|Y';
--
\set ON_ERROR_STOP on

-- Take input from the user.
--
\prompt 'Are you sure to drop ' Do_You_Want_To_Drop_Db

-- Get the user input.
--
\set Do_Drop_Db '\'' :Do_You_Want_To_Drop_Db '\''

-- Creating a temp table to store the value of the user confirmation.
-- This will be dropped when the session got closed.
--
CREATE TEMP TABLE Drop_Db AS SELECT :Do_Drop_Db::text AS confirm;

DO
$$
BEGIN
IF EXISTS(SELECT * FROM Drop_Db WHERE confirm NOT IN('y', 'Y')) THEN
RAISE EXCEPTION 'Database won''t drop. Hence closing this session';
ELSE
RAISE NOTICE 'Database will be droped. Please make sure the connection of this databases get closed.';
END IF;
END
$$;

DROP DATABASE sample;



Testing the Script
bash-4.1$ ../bin/psql -U postgres -f /tmp/Interactive_Sql.Sql postgres
Are you sure to drop N
SELECT 1
psql:/tmp/Interactive_Sql.Sql:28: ERROR:  Database won't drop. Hence closing this session
bash-4.1$ ../bin/psql -U postgres -f /tmp/Interactive_Sql.Sql postgres
Are you sure to drop Y
SELECT 1
psql:/tmp/Interactive_Sql.Sql:28: NOTICE:  Database will be droped. Please make sure the connection of this databases get closed.
DO
DROP DATABASE

దినేష్ కుమార్ 
Dinesh Kumar

Comments

  1. Thank u for the useful script and I was searching for this style of script in postgres.

    ReplyDelete
  2. You don't know how long have I been looking for a way to get interactive input. Thank you!

    ReplyDelete

Post a Comment

Popular posts from this blog

Parallel Operations With pl/pgSQL

Hi, I am pretty sure that, there will be a right heading for this post. For now, i am going with this. If you could suggest me proper heading, i will update it :-) OK. let me explain the situation. Then will let you know what i am trying to do here, and how i did it. Situation here is, We have a table, which we need to run update on “R” no.of records. The update query is using some joins to get the desired result, and do update the table.  To process these “R” no.of records, it is taking “H” no.of hours. That too, it’s giving load on the production server. So, we planned to run this UPDATE as batch process.  Per a batch process, we took “N” no.or records. To process this batch UPDATE, it is taking “S” no.of seconds. With the above batch process, production server is pretty stable, and doing great. So, we planned to run these Batch updates parallel.  I mean, “K” sessions, running different record UPDATEs. Of-course, we can also increase the Batch size here.  But

How To Send E-Mail From PostgreSQL

Hi , If you want to send E-Mails from PostgreSQL, then use the below Python 3.2 Script as below. I have used ActivePython 3.2 with PostgreSQL 9.1 for sending E-Mails from PostgreSQL. If you want to configure the Python 3.2 with PostgreSQL 9.1 then, please refer the below steps. http://manojadinesh.blogspot.in/2012/06/fatal-python-error-pyinitialize-unable.html Once, your Python 3.2 successful then follow the below steps to send an e-mail. Step 1 ===== postgres=# CREATE OR REPLACE FUNCTION public.send_email(_from Text,_password Text,smtp Text,port INT,receiver text, subject text, send_message text) RETURNS TEXT  LANGUAGE plpython3u AS $function$ import smtplib sender = _from receivers = receiver message = ("From: %s\nTo: %s\nSubject: %s\n\n %s"  % (_from,receiver,subject,send_message)) try:   smtpObj = smtplib.SMTP(smtp,port)   smtpObj.starttls()   smtpObj.login(_from, _password)   smtpObj.sendmail(sender, receivers,message)   print ('Successf

::Pipelined in Oracle as well in PostgreSQL::

Pipelined Table Functions:- [ORACLE] =========================== If you want to return multiple rows to the calling environment, then piplined table functions is prefred. It will increase the dbperformance as well. Ex:- Step 1: ----------- CREATE TABLE EMP(EMPNO INT,ENAME VARCHAR2(10),SAL INT); Step 2: ----------- Insert sample data. Step 3: ----------- Create an object for the row type casting. CREATE OR REPLACE TYPE emp_row AS OBJECT ( empno INT, ename VARCHAR2(20), SAL INT ); Step 4: ----------- Create a Return Type for the pipelined function. CREATE OR REPLACE TYPE emp_table_type AS TABLE OF emp_row; Step 5: ----------- CREATE OR REPLACE FUNCTION emp_pipe_function RETURN emp_table_type PIPELINED IS BEGIN FOR rec in (select * from emp) LOOP PIPE ROW (emp_row(rec.empno,rec.ename,rec.sal)); END LOOP; RETURN; END; Step 6: ---------- SQL> select * from table(emp_pipe_function); EMPNO ENAME SAL ---------- ----