7bYZNaBv

1import sqlite3 as sql
2import os
3import csv
4from sqlite3 import Error
5
6try:
7
8  # Connect to database
9  conn=sql.connect('mydb.db')
10
11  # Create Table into database
12  conn.execute('''CREATE TABLE IF NOT EXISTS Employee(Id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,\
13               Name TEXT NOT NULL, Salary INT NOT NULL 
14              );''')
15  # Insert some values to database
16  conn.execute('''INSERT INTO Employee(Name, Salary) VALUES('Laxmi', 30000);''')
17  conn.execute('''INSERT INTO Employee(Name, Salary) VALUES('Prerna', 40000);''')
18  conn.execute('''INSERT INTO Employee(Name, Salary) VALUES('Shweta', 30000);''')
19  conn.execute('''INSERT INTO Employee(Name, Salary) VALUES('Soniya', 50000);''')
20  conn.execute('''INSERT INTO Employee(Name, Salary) VALUES('Priya', 60000);''')
21  conn.commit()
22
23 # To view table data in table format
24  print "******Employee Table Data*******"
25  cur = conn.cursor()
26  cur.execute('''SELECT * FROM Employee''')
27  rows = cur.fetchall()
28   
29  for row in rows:
30      print(row)
31
32 # Export data into CSV file
33  print "Exporting data into CSV............"
34  cursor = conn.cursor()
35  cursor.execute("select * from Employee")
36  with open("employee_data.csv", "w") as csv_file:
37      csv_writer = csv.writer(csv_file, delimiter="\t")
38      csv_writer.writerow([i[0] for i in cursor.description])
39      csv_writer.writerows(cursor)
40
41  dirpath = os.getcwd() + "/employee_data.csv"
42  print "Data exported Successfully into {}".format(dirpath)
43
44except Error as e:
45  print(e)
46
47# Close database connection
48finally:
49  conn.close()