def assign_command(n=-1, p=-1, t=-1): """ Assign a note to a project or task, or a task to a project. """ c = conn() if n > 0: # We are assigning a note to a project or task. if p > 0: element = Project.get(c, p) elif t > 0: element = Task.get(c, t) else: raise Exception("You must specify either a project id or a task id.") note = Note.get(c, n) note.assign(c, element) elif t > 0: task = Task.get(c, t) # We are assigning a task to a project. if p > 0: project = Project.get(c, p) task.assign(c, project) else: raise Exception("You must specify a project id to assign the task to.") else: raise Exception("You didn't specify anything to assign!")
def project_command( name=None, # the name of the project (required) description="", # an optional description for this project p=-1, # portfolio id for this project (required unless parent project specified) parent=-1, # parent project id, if this is a subproject ): """ Create a new project. """ c = conn() if parent > 0: parent_project_id = parent parent_project = Project.get(c, parent) portfolio_id = parent_project.portfolio_id else: parent_project_id = None if p > 0: portfolio_id = p else: raise Exception( "You must provide a portfolo id using the -p parameter if this project doesn't have a parent project." ) project = Project.create( c, created_at=datetime.now(), description=description, parent_project_id=parent_project_id, name=name, portfolio_id=portfolio_id, ) print project.id,
def test_create_table(): conn = Model.setup_db() Project.create_table(conn) sql = "select * from sqlite_master where type='table' and name='Project'" assert len(conn.execute(sql).fetchall()) == 1 row = conn.execute(sql).fetchone() assert row['sql'] == Project.create_table_sql() conn.execute("select * from Project")
def test_create_project(): conn = get_conn() project = Project.create(conn, name="My New Project", created_at = datetime.datetime.now()) assert project.id == 1 assert project.elapsed_seconds() < 0.01 lookup_project = Project.get(conn, 1) assert lookup_project.name == "My New Project" assert project.elapsed_seconds() < 0.01
def archive_command( n=-1, p=-1, t=-1 # id of the note to archive # id of the project to archive # id of the task to archive ): """ Archive the note, project or task specified. """ c = conn() if n > 0: Note.archive(c, n) elif p > 0: Project.archive(c, p) elif t > 0: Task.archive(c, t) else: raise Exception()
def delete_command( n=-1, p=-1, t=-1 # id of the note to delete # id of the project to delete # id of the task to delete ): """ Delete the note, project or task specified. """ c = conn() if n > 0: Note.delete(c, n) elif p > 0: Project.delete(c, p) elif t > 0: Task.delete(c, t) else: raise Exception()
def update_command(t=-1, p=-1, n=-1, r=-1, **kwargs): """ Update a project, task or note with the supplied kwargs. """ c = conn() if t > 0: Task.update(c, t, kwargs) elif n > 0: Note.update(c, n, kwargs) elif p > 0: Project.update(c, p, kwargs) elif r > 0: Recipe.update(c, r, kwargs) else: raise Exception("Must specify one of t (task), n (note), p (project) or r (recipe).")
def test_create_note(): conn = get_conn() note = Note.create(conn, note="This is a new note.", created_at = datetime.datetime.now()) assert note.id == 1 assert note.elapsed_seconds() < 0.01 project = Project.create( conn, name = "My Project With Notes", created_at = datetime.datetime.now() ) note2 = Note.create( conn, note="This is a note assigned to a project.", linked_to_type="Project", linked_to_id=project.id, created_at=datetime.datetime.now() ) assert note2.id == 2 inbox_notes = [n.id for n in Note.inbox(conn)] assert note.id in inbox_notes assert not note2.id in inbox_notes note.assign(conn, project) inbox_notes = [n.id for n in Note.inbox(conn)] assert not note.id in inbox_notes assert not note2.id in inbox_notes note = note.reload(conn) assert note.project(conn).id == project.id assert note2.project(conn).id == project.id
def complete_command(p=-1, t=-1): # id of the project to mark complete # id of the task to mark complete """ Mark the project or task as completed. """ c = conn() if p > 0: project = Project.get(c, p) project.complete(c) Project.archive(c, project.id) print "Project %s marked as complete!" % p elif t > 0: task = Task.get(c, t) task.complete(c) print "Task %s marked as complete!" % t else: raise Exception()
def note_command( note="", # the contents of the note p=-1, # the project id to link the new note to (optional) t=-1, # the task id to link the new note to (optional) ): """ Create a new note. If note contents are not specified, will read from STDIN. """ c = conn() if len(note) == 0: note = sys.stdin.read().strip() if len(note) == 0: raise Exception("You didn't pass any content for your note!") n = Note.create(c, note=note, created_at=datetime.now()) print "Created note", n.id if p > 0: project = Project.get(c, p) n.assign(c, project) print "Assigned to project %s" % p elif t > 0: task = Task.get(c, t) n.assign(c, task) print "Assigned to task %s" % t
def projects_command(): """ List all projects. """ c = conn() projects = Project.all_nested_subprojects(c) for project in projects: print project.display_line() if len(projects) == 0: print "No projects found. Run 'ado help' if you need help."
def show_command( t=-1, # id of task to show detail on n=-1, # id of note to show detail on p=-1, # id of project to show detail on portfolio=-1, # id of portfolio to show detail on ): """ Print detailed information for a project, task, note or portfolio. """ c = conn() if t > 0: task = Task.get(c, t) print task.show() elif n > 0: note = Note.get(c, n) print note.show() elif p > 0: project = Project.get(c, p) print project.show() elif portfolio > 0: portfolio = Portfolio.get(c, portfolio) print portfolio.show() else: raise Exception("Must specify one of t (task), n (note), p (project) or portfolio.")
def projects(self, conn=None): if not conn: conn = self.conn sql = "select * from %s where portfolio_id = ? and archived_at IS NULL" % Project.table_name() return [Project.load(conn, row) for row in conn.execute(sql, [self.id])]
def projects(self): return Project.all_nested_subprojects(conn())
def test_table_name(): assert Model.table_name() == "Model" assert Project.table_name() == "Project"
def test_persist_instance(): conn = Model.setup_db() Project.create_table(conn) project = Project.create(conn, name="My New Project") print project.persist_instance_sql() print project.persist(conn)