-
Notifications
You must be signed in to change notification settings - Fork 0
/
databasesql.py
166 lines (104 loc) · 3.81 KB
/
databasesql.py
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
163
164
165
# custom module "DATABASESQL" that creates functions that contain sql statements that can be used to stores and retrieves data from
# A database
#this is an independent Module That will be imported in the logic.py but with sql statements
#Author:Akola Arthur Alaali
#Contact Information
#Email:arthurstaurtleo@hotmail.com
# :arthurstaurtleo@gmail.com
#phone:+256758855695
#created On:19th June 2018 and Modified On:20th June 2018
import sqlite3 as sql,sys
#sqlite inbuilt functions connect(),cursor(),commit()
FNAME= "smschat.db"
def init():
db = sql.connect(FNAME)
cur=db.cursor()
stmt = """ create table if not exists user_details(
User_id int primary key not null,
Username varchar(25) not null,
Time_sent_in_seconds real not null
)"""
cur.execute(stmt)
db.commit()
db = sql.connect(FNAME)
cur=db.cursor()
messages =""" create table if not exists messages(
message_to int primary key not null,
message_from varchar(25) not null,
Text varchar(100),
Time_sent_human_form varchar(30) not null,
Time_sent_in_seconds real not null
)"""
cur.execute(messages)
db.commit()
return db, cur
def add_user(db,cur,new_id,uname,Time_sent_in_seconds):
cur=db.cursor()
stmt="""
insert into user_details values(?,?,?)
"""
try:
cur.execute( stmt, (new_id,uname,Time_sent_in_seconds))
db.commit()
return True
except:
print "Unable to add User to the Database"
print ""
print "The User_Id Entered Already Exists In The System"#
return False
def read_user(db,cur):#,uname not required
cur.execute("select Username from user_details")
users_in=cur.fetchall()
return [user_in[0] for user_in in users_in]
#print returned_users
#print users_in
#return users_in
def read_all_users(db,cur):#,uname not required
cur.execute("select Username from user_details")
all_users_in=cur.fetchall()#thi
print all_users_in
return all_users_in #[ user_in[0] for user_in in users_in]
def save_message(db,cur,message_to,message_from,Text,Time_sent_human_form):#*read_all_user()
cur=db.cursor()#points to a pointer in the database file
msg="""
insert into messages values(?,?,?,?,?)
"""
msg=cur.execute( msg, (message_to,message_from,Text,time.acstime(),time.time()) )
db.commit()
# print msg
#def read_inbox(db,cur,user_id):
# cur.execute("""select User_id,Time_sent_in_seconds from user_details where User_id = ?""",(user_id,))
# specific_user_id=cur.fetchall()
# cur.execute("select message_from,Text,Time_sent_human_form from messages where User_id = ?",(user_id,))
#message_for_user_id = cur.fetchall()
def get_inbox(db, cur, uid):
cur.execute("select last_fetch_time from users_details where ID = ?",(uid,))
last_fetch_time = cur.fetchone()
if not last_fetch_time:
return []
last_fetch_time = last_fetch_time[0]
t=time.time()
cur.execute("""
select time_sent_human_form,message_from,message from messages where
(message_to = ?) and (message_time_in_seconds > ?)""",
(uid,last_fetch_time)
)
messages = cur.fetchall()
cur.execute("update users_details set last_fetch_time=? where ID=?",(t,uid))
db.commit()
return messages
#if (Time_sent_in_seconds
#def read_msg_data:
db,cur=init()
#add_user(db,cur,"1011","Mush",0.0)
read_user(db,cur)#,"daisy"
#users =read_user(*init())
#print users
#db,cur=init()
#save_message(db,cur,10013,"bug","Thank You,we..","Wed Jun 20 10:42:19 2018")
#db,cur =init()
#read_inbox(db,cur,10012)
def g():
for i in range(20):
yield i
print g().next()#