add createTask(), getNextId()
This commit is contained in:
parent
951bf59dab
commit
66f2ee7c17
28
DataStore.py
28
DataStore.py
@ -59,10 +59,36 @@ FROM projects
|
|||||||
SELECT id, name
|
SELECT id, name
|
||||||
FROM tasks
|
FROM tasks
|
||||||
WHERE projectid = ?
|
WHERE projectid = ?
|
||||||
''' % (projectid,))
|
''', (projectid,))
|
||||||
tasks = {}
|
tasks = {}
|
||||||
for row in c:
|
for row in c:
|
||||||
tasks[row[0]] = row[1]
|
tasks[row[0]] = row[1]
|
||||||
c.close()
|
c.close()
|
||||||
self.conn.commit()
|
self.conn.commit()
|
||||||
return tasks
|
return tasks
|
||||||
|
|
||||||
|
def createTask(self, projectid, taskname):
|
||||||
|
c = self.conn.cursor()
|
||||||
|
tasks = self.getTasks(projectid)
|
||||||
|
if taskname in tasks.values():
|
||||||
|
return False
|
||||||
|
nextid = self.getNextId('tasks')
|
||||||
|
c.execute('''
|
||||||
|
INSERT INTO tasks (id, projectid, name)
|
||||||
|
VALUES (?, ?)
|
||||||
|
''', (nextid, projectid, taskname))
|
||||||
|
c.close()
|
||||||
|
self.conn.commit()
|
||||||
|
return True
|
||||||
|
|
||||||
|
def getNextId(self, table):
|
||||||
|
c = self.conn.cursor()
|
||||||
|
nextid = 1
|
||||||
|
c.execute('''
|
||||||
|
SELECT MAX(id)
|
||||||
|
FROM %s
|
||||||
|
''' % table)
|
||||||
|
for row in c:
|
||||||
|
nextid = row[0] + 1
|
||||||
|
c.close()
|
||||||
|
return nextid
|
||||||
|
Loading…
x
Reference in New Issue
Block a user