This repository was archived by the owner on Jul 18, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 28
Expand file tree
/
Copy pathibm_db-rollback.py
More file actions
162 lines (141 loc) · 8.36 KB
/
ibm_db-rollback.py
File metadata and controls
162 lines (141 loc) · 8.36 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
#! /usr/bin/python3
#-------------------------------------------------------------------------------------------------#
# NAME: ibm_db-rollback.py #
# #
# PURPOSE: This program is designed to illustrate how to use the ibm_db.rollback() API. #
# #
# Additional APIs used: #
# ibm_db.autocommit() #
# ibm_db.exec_immediate() #
# ibm_db.stmt_errormsg() #
# #
# USAGE: Log in as a Db2 database instance user (for example, db2inst1) and issue the #
# following command from a terminal window: #
# #
# ./ibm_db-rollback.py #
# #
#-------------------------------------------------------------------------------------------------#
# DISCLAIMER OF WARRANTIES AND LIMITATION OF LIABILITY #
# #
# (C) COPYRIGHT International Business Machines Corp. 2018, 2019 All Rights Reserved #
# Licensed Materials - Property of IBM #
# #
# US Government Users Restricted Rights - Use, duplication or disclosure restricted by GSA ADP #
# Schedule Contract with IBM Corp. #
# #
# The following source code ("Sample") is owned by International Business Machines Corporation #
# or one of its subsidiaries ("IBM") and is copyrighted and licensed, not sold. You may use, #
# copy, modify, and distribute the Sample in any form without payment to IBM, for the purpose #
# of assisting you in the creation of Python applications using the ibm_db library. #
# #
# The Sample code is provided to you on an "AS IS" basis, without warranty of any kind. IBM #
# HEREBY EXPRESSLY DISCLAIMS ALL WARRANTIES, EITHER EXPRESS OR IMPLIED, INCLUDING, BUT NOT #
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. #
# Some jurisdictions do not allow for the exclusion or limitation of implied warranties, so the #
# above limitations or exclusions may not apply to you. IBM shall not be liable for any damages #
# you suffer as a result of using, copying, modifying or distributing the Sample, even if IBM #
# has been advised of the possibility of such damages. #
#-------------------------------------------------------------------------------------------------#
# Load The Appropriate Python Modules
import sys # Provides Information About Python Interpreter Constants, Functions, & Methods
import ibm_db # Contains The APIs Needed To Work With Db2 Databases
#-------------------------------------------------------------------------------------------------#
# Import The Db2ConnectionMgr Class Definition, Attributes, And Methods That Have Been Defined #
# In The File Named "ibm_db_tools.py"; This Class Contains The Programming Logic Needed To #
# Establish And Terminate A Connection To A Db2 Server Or Database #
#-------------------------------------------------------------------------------------------------#
from ibm_db_tools import Db2ConnectionMgr
#-------------------------------------------------------------------------------------------------#
# Import The get_row_count() Function That Has Been Defined In The File Named "ibm_db_tools.py"; #
# This Function Contains The Programming Logic Needed To Obtain And Display The Number Of Rows #
# (Records) Found In A Db2 Database Table. #
#-------------------------------------------------------------------------------------------------#
from ibm_db_tools import get_row_count
#-------------------------------------------------------------------------------------------------#
# Import The ipynb_exit Class Definition, Attributes, And Methods That Have Been Defined In The #
# File Named "ipynb_exit.py"; This Class Contains The Programming Logic Needed To Allow "exit()" #
# Functionality To Work Without Raising An Error Or Stopping The Kernel If The Application Is #
# Invoked In A Jupyter Notebook #
#-------------------------------------------------------------------------------------------------#
from ipynb_exit import exit
# Define And Initialize The Appropriate Variables
dbName = "SAMPLE"
userID = "db2inst1"
passWord = "Passw0rd"
dbConnection = None
returnCode = False
resultSet = False
# Create An Instance Of The Db2ConnectionMgr Class And Use It To Connect To A Db2 Database
conn = Db2ConnectionMgr('DB', dbName, '', '', userID, passWord)
conn.openConnection()
if conn.returnCode is True:
dbConnection = conn.connectionID
else:
conn.closeConnection()
exit(-1)
# Turn Autocommit Behavior OFF
print("Turning AUTOCOMMIT behavior OFF ... ", end="")
try:
returnCode = ibm_db.autocommit(dbConnection, ibm_db.SQL_AUTOCOMMIT_OFF)
except Exception:
pass
# If AUTOCOMMIT Behavior Could Not Be Turned OFF, Display An Error Message And Continue
if returnCode is False:
print("\nERROR: Unable to turn AUTOCOMMIT behavior OFF.")
# Otherwise, Complete The Status Message
else:
print("Done!\n")
# Display The Number Of Rows That Currently Exist In The DEPARTMENT Table
returnCode = get_row_count(dbConnection, 'DEPARTMENT')
if returnCode is False:
conn.closeConnection()
exit(-1)
# Define The INSERT Statement That Is To Be Used To Add Data To The DEPARTMENT Table
sqlStatement = "INSERT INTO department VALUES('K01', 'SALES', '000130', 'K01', NULL)"
# Execute The SQL Statement Just Defined
print("Inserting a record into the DEPARTMENT table ... ", end="")
try:
resultSet = ibm_db.exec_immediate(dbConnection, sqlStatement)
except Exception:
pass
# If The SQL Statement Could Not Be Executed, Display An Error Message And Exit
if resultSet is False:
print("\nERROR: Unable to execute the INSERT statement specified.")
errorMsg = ibm_db.stmt_errormsg()
print("\n" + errorMsg + "\n")
conn.closeConnection()
exit(-1)
# Otherwise, Complete The Status Message
else:
print("Done!\n")
# Display The Number Of Rows That Exist In The DEPARTMENT Table Now
# (The Number Returned Should Change)
returnCode = get_row_count(dbConnection, 'DEPARTMENT')
if returnCode is False:
conn.closeConnection()
exit(-1)
# Back Out The Changes Just Made To The Database
print("Backing out changes made to the database ... ", end="")
resultSet = False
try:
resultSet = ibm_db.rollback(dbConnection)
except Exception:
pass
# If The Rollback Operation Could Not Be Performed, Display An Error Message And Exit
if resultSet is False:
print("\nERROR: Unable to roll back the previous operation.")
conn.closeConnection()
exit(-1)
# Otherwise, Complete The Status Message
else:
print("Done!\n")
# Display The Number Of Rows That Exist In The DEPARTMENT Table Now
# (The Number Should Revert Back To The Original Value)
returnCode = get_row_count(dbConnection, 'DEPARTMENT')
if returnCode is False:
conn.closeConnection()
exit(-1)
# Close The Database Connection That Was Opened Earlier
conn.closeConnection()
# Return Control To The Operating System
exit()