add createTask(), getNextId()

This commit is contained in:
Josh Holtrop 2011-02-15 22:15:21 -05:00
parent 951bf59dab
commit 66f2ee7c17

View File

@ -59,10 +59,36 @@ FROM projects
SELECT id, name
FROM tasks
WHERE projectid = ?
''' % (projectid,))
''', (projectid,))
tasks = {}
for row in c:
tasks[row[0]] = row[1]
c.close()
self.conn.commit()
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