fz0x1/jrnl
1
0
Fork 0
mirror of https://github.com/jrnl-org/jrnl.git synced 2025-05-10 16:48:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Formatting

Browse source
This commit is contained in:
Manuel Ebert 2013-04-23 09:13:06 -07:00
parent 4528d9f612
commit 1417305c2e
1 changed files with 17 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
jrnl/Journal.py
View file

@ -31,6 +31,7 @@ except ImportError:
import plistlib import plistlib
import uuid import uuid
class Journal(object): class Journal(object):
def __init__(self, **kwargs): def __init__(self, **kwargs):
self.config = { self.config = {
@ -48,9 +49,9 @@ class Journal(object):
# Set up date parser # Set up date parser
consts = pdt.Constants() consts = pdt.Constants()
consts.DOWParseStyle = -1 # "Monday" will be either today or the last Monday consts.DOWParseStyle = -1 # "Monday" will be either today or the last Monday
self.dateparse = pdt.Calendar(consts) self.dateparse = pdt.Calendar(consts)
self.key = None # used to decrypt and encrypt the journal self.key = None # used to decrypt and encrypt the journal
journal_txt = self.open() journal_txt = self.open()
self.entries = self.parse(journal_txt) self.entries = self.parse(journal_txt)
@ -74,7 +75,7 @@ class Journal(object):
except ValueError: except ValueError:
print("ERROR: Your journal file seems to be corrupted. You do have a backup, don't you?") print("ERROR: Your journal file seems to be corrupted. You do have a backup, don't you?")
sys.exit(-1) sys.exit(-1)
if plain[-1] != " ": # Journals are always padded if plain[-1] != " ": # Journals are always padded
return None return None
else: else:
return plain return plain
@ -83,12 +84,12 @@ class Journal(object):
"""Encrypt a plaintext string using self.key as the key""" """Encrypt a plaintext string using self.key as the key"""
if not crypto_installed: if not crypto_installed:
sys.exit("Error: PyCrypto is not installed.") sys.exit("Error: PyCrypto is not installed.")
atfork() # A seed for PyCrypto atfork() # A seed for PyCrypto
iv = ''.join(chr(random.randint(0, 0xFF)) for i in range(16)) iv = ''.join(chr(random.randint(0, 0xFF)) for i in range(16))
crypto = AES.new(self.key, AES.MODE_CBC, iv) crypto = AES.new(self.key, AES.MODE_CBC, iv)
if len(plain) % 16 != 0: if len(plain) % 16 != 0:
plain += " " * (16 - len(plain) % 16) plain += " " * (16 - len(plain) % 16)
else: # Always pad so we can detect properly decrypted files :) else: # Always pad so we can detect properly decrypted files :)
plain += " " * 16 plain += " " * 16
return iv + crypto.encrypt(plain) return iv + crypto.encrypt(plain)
@ -112,7 +113,7 @@ class Journal(object):
decrypted = self._decrypt(journal) decrypted = self._decrypt(journal)
if decrypted is None: if decrypted is None:
attempts += 1 attempts += 1
self.config['password'] = None # This password doesn't work. self.config['password'] = None # This password doesn't work.
if attempts < 3: if attempts < 3:
print("Wrong password, try again.") print("Wrong password, try again.")
else: else:
@ -160,7 +161,7 @@ class Journal(object):
"""Prettyprints the journal's entries""" """Prettyprints the journal's entries"""
sep = "\n" sep = "\n"
pp = sep.join([e.pprint() for e in self.entries]) pp = sep.join([e.pprint() for e in self.entries])
if self.config['highlight']: # highlight tags if self.config['highlight']: # highlight tags
if self.search_tags: if self.search_tags:
for tag in self.search_tags: for tag in self.search_tags:
tagre = re.compile(re.escape(tag), re.IGNORECASE) tagre = re.compile(re.escape(tag), re.IGNORECASE)
@ -176,17 +177,17 @@ class Journal(object):
def __repr__(self): def __repr__(self):
return "<Journal with %d entries>" % len(self.entries) return "<Journal with %d entries>" % len(self.entries)
def write(self, filename = None): def write(self, filename=None):
"""Dumps the journal into the config file, overwriting it""" """Dumps the journal into the config file, overwriting it"""
filename = filename or self.config['journal'] filename = filename or self.config['journal']
journal = "\n".join([str(e) for e in self.entries]) journal = "\n".join([str(e) for e in self.entries])
if self.config['encrypt']: if self.config['encrypt']:
journal = self._encrypt(journal) journal = self._encrypt(journal)
with open(filename, 'wb') as journal_file: with open(filename, 'wb') as journal_file:
journal_file.write(journal) journal_file.write(journal)
else: else:
with codecs.open(filename, 'w', "utf-8") as journal_file: with codecs.open(filename, 'w', "utf-8") as journal_file:
journal_file.write(journal) journal_file.write(journal)
def sort(self): def sort(self):
"""Sorts the Journal's entries by date""" """Sorts the Journal's entries by date"""
@ -243,10 +244,10 @@ class Journal(object):
date, flag = self.dateparse.parse(date) date, flag = self.dateparse.parse(date)
if not flag: # Oops, unparsable. if not flag: # Oops, unparsable.
return None return None
if flag is 1: # Date found, but no time. Use the default time. if flag is 1: # Date found, but no time. Use the default time.
date = datetime(*date[:3], hour=self.config['default_hour'], minute=self.config['default_minute']) date = datetime(*date[:3], hour=self.config['default_hour'], minute=self.config['default_minute'])
else: else:
date = datetime(*date[:6]) date = datetime(*date[:6])
@ -268,7 +269,7 @@ class Journal(object):
# Split raw text into title and body # Split raw text into title and body
title_end = len(raw) title_end = len(raw)
for separator in ["\n",". ","? ","! "]: for separator in ["\n", ". ", "? ", "! "]:
sep_pos = raw.find(separator) sep_pos = raw.find(separator)
if 1 < sep_pos < title_end: if 1 < sep_pos < title_end:
title_end = sep_pos title_end = sep_pos
@ -277,9 +278,9 @@ class Journal(object):
if not date: if not date:
if title.find(":") > 0: if title.find(":") > 0:
date = self.parse_date(title[:title.find(":")]) date = self.parse_date(title[:title.find(":")])
if date: # Parsed successfully, strip that from the raw text if date: # Parsed successfully, strip that from the raw text
title = title[title.find(":")+1:].strip() title = title[title.find(":")+1:].strip()
if not date: # Still nothing? Meh, just live in the moment. if not date: # Still nothing? Meh, just live in the moment.
date = self.parse_date("now") date = self.parse_date("now")
entry = Entry.Entry(self, date, title, body) entry = Entry.Entry(self, date, title, body)
@ -288,6 +289,7 @@ class Journal(object):
self.sort() self.sort()
return entry return entry
class DayOne(Journal): class DayOne(Journal):
"""A special Journal handling DayOne files""" """A special Journal handling DayOne files"""
def __init__(self, **kwargs): def __init__(self, **kwargs):
@ -314,7 +316,6 @@ class DayOne(Journal):
# we're returning the obvious. # we're returning the obvious.
return self.entries return self.entries
def write(self): def write(self):
"""Writes only the entries that have been modified into plist files.""" """Writes only the entries that have been modified into plist files."""
for entry in self.entries: for entry in self.entries:
@ -331,4 +332,3 @@ class DayOne(Journal):
'UUID': new_uuid 'UUID': new_uuid
} }
plistlib.writePlist(entry_plist, filename) plistlib.writePlist(entry_plist, filename)