class TestFromPitzdir(unittest.TestCase):
def setUp(self):
self.p = Project(pathname='/tmp')
self.p.to_yaml_file()
self.p.to_pickle()
def tearDown(self):
for f in glob.glob('/tmp/*.yaml'):
os.unlink(f)
if os.path.isfile('/tmp/project.pickle'):
os.unlink('/tmp/project.pickle')
def test_fresh_pickle(self):
"""
Verify we use the pickle when we can.
"""
p = Project.from_pitzdir('/tmp')
assert p.loaded_from == 'pickle', p.loaded_from
def test_stale_pickle(self):
"""
Verify we use the yaml files when the pickle is too old.
"""
stat = os.stat('/tmp/project.pickle')
os.utime('/tmp/project.pickle',
(stat.st_atime-1, stat.st_mtime-1))
print("pickle file: %s"
% os.stat('/tmp/project.pickle').st_mtime)
print("newest yaml file: %s"
% max([os.stat(f).st_mtime for f in glob.glob('/tmp/*.yaml')]))
p = Project.from_pitzdir('/tmp')
assert p.loaded_from == 'yaml', p.loaded_from
def test_from_yaml_files(self):
"""
Verify we can use the yaml files when no pickle exists.
"""
os.unlink('/tmp/project.pickle')
p = Project.from_pitzdir('/tmp')
assert p.loaded_from == 'yaml', p.loaded_from
def pitz_setup():
"""
Start a new project
"""
p = optparse.OptionParser()
p.epilog = "Set up pitz"
p.add_option('--version', action='store_true',
help='show pitz version')
options, args = p.parse_args()
if options.version:
print_version()
return
dir = os.path.basename(os.getcwd())
project_name = raw_input(
"Project name (hit ENTER for %s): " % dir).strip()
if not project_name:
project_name = dir
pitzdir = mk_pitzdir()
proj = Project(pathname=pitzdir, title=project_name)
proj.to_yaml_file()
Status.setup_defaults(proj)
pw_name = pwd.getpwuid(os.getuid()).pw_name
name = raw_input("Your name (hit ENTER for %s): " % pw_name).strip()
if not name:
name = pw_name
person = Person(proj, title=name)
proj.save_entities_to_yaml_files()
print("All done!")
print("Run pitz-add-task to add a task, or run pitz-help for help.")
def test_to_yaml_file_2(o):
p = Project("Bogus")
p.to_yaml_file()
def test_to_yaml_file_1(m1, m2):
p = Project("Bogus")
p.to_yaml_file('bogus-pathname')