1
+ import sqlite3
2
+ import os
3
+ from sqlite3 import Error
4
+
5
+ def create_connection (Education ):
6
+ conn = None
7
+ try :
8
+ conn = sqlite3 .connect (Education )
9
+ except Error as e :
10
+ print (e )
11
+
12
+ return conn
13
+
14
+ # col_name
15
+ def select_task (conn ,table_name ,col_name ,where_clause ):
16
+ cur = conn .cursor ()
17
+ #new string
18
+ str_query = ""
19
+
20
+ str_query = str_query + "SELECT "
21
+ print ("After adding select keyword " ,str_query )
22
+
23
+ #string for all the relevent columns of the table:
24
+ str_column_list = ""
25
+ # SELECT Roll_Number,a,b,c FROM Student
26
+ for i in range (len (col_name )):
27
+ str_column_list = str_column_list + col_name [i ]
28
+ print (str_column_list )
29
+ if i != len (col_name )- 1 :
30
+ str_column_list = str_column_list + ","
31
+
32
+ print ( " Column Names are " ,str_column_list )
33
+
34
+ str_query = str_query + str_column_list
35
+ print ( " Now Query is " ,str_query )
36
+
37
+ str_query = str_query + " FROM " + table_name + " "
38
+
39
+ print ( " Now Query is " ,str_query )
40
+
41
+ if where_clause != "" :
42
+ str_query = str_query + " WHERE " + where_clause
43
+
44
+ print ( " Now Query is " ,str_query )
45
+ cur .execute (str_query )
46
+ rows = cur .fetchall ()
47
+ for row in rows :
48
+ print (row )
49
+
50
+ def delete_task (conn ,table_name ,where_clause ):
51
+ cur = conn .cursor ()
52
+ #DELETE FROM Student where Subject_ID=2
53
+ str_query = ""
54
+ if where_clause != "" :
55
+ str_query = str_query + "DELETE FROM " + table_name + " WHERE " + where_clause
56
+ else :
57
+ str_query = str_query + "DELETE FROM " + table_name
58
+
59
+ str_query = "Select * FROM " + table_name
60
+ print ( " Now Delete Query is " ,str_query )
61
+
62
+ cur .execute (str_query )
63
+
64
+ rows = cur .fetchall ()
65
+
66
+ for row in rows :
67
+ print (row )
68
+
69
+ # update tb set c1 = c1_val , c2 = c2_val WHERE c_name = c_name_value
70
+ def update_task (conn ,table_name ,col_name ,data ,where_clause ):
71
+ cur = conn .cursor ()
72
+ str_query = ""
73
+ str_query = str_query + "update " + table_name + " set "
74
+
75
+ for i in range (len (col_name )):
76
+ str_query = str_query + col_name [i ] + " = '" + data [i ] + "'"
77
+ if i != len (col_name )- 1 :
78
+ str_query = str_query + ","
79
+
80
+ if where_clause != "" :
81
+ str_query = str_query + " WHERE " + where_clause
82
+ #str = "update Subject set branch='computer science' where subject_id = 3"
83
+ print ( " Now Update Query is " ,str_query )
84
+
85
+ cur .execute (str_query )
86
+
87
+ rows = cur .fetchall ()
88
+
89
+ for row in rows :
90
+ print (row )
91
+
92
+
93
+
94
+ def main ():
95
+ database = r"Education.db"
96
+ print ("Current Path : " ,os .getcwd ())
97
+ conn = create_connection (database )
98
+ with conn :
99
+ print ("1. Query task" )
100
+
101
+ col_name = ['Roll_Number' ,'Name' ]
102
+ #select_task(conn,'Student',col_name,"Name='Ram'")
103
+ select_task (conn ,'Student' ,col_name ,"" )
104
+
105
+ #col_name = ['Branch','Subject_Name']
106
+ delete_task (conn ,'Subject' ,"Subject_ID=4" )
107
+
108
+ col_name = ['Name' ,'Address' ,'Id_Proof' ]
109
+ data_send = ['Astha' ,'Gorakhpur' ,'PAN' ]
110
+ update_task (conn ,'Student' ,col_name ,data_send ,"Roll_Number=3" )
111
+
112
+ col_name = ['Name' ,'Roll_Number' ,'Address' ,'Phone_Number' ,'Id_Proof' ]
113
+ #select_task(conn,'Student',col_name,"Name='Ram'")
114
+ select_task (conn ,'Student' ,col_name ,"" )
115
+
116
+
117
+ if __name__ == '__main__' :
118
+ main ()
0 commit comments