diff options
-rw-r--r-- | lesana/command.py | 66 | ||||
-rwxr-xr-x | scripts/lesana | 3 |
2 files changed, 68 insertions, 1 deletions
diff --git a/lesana/command.py b/lesana/command.py index 23b9b6b..706bbf3 100644 --- a/lesana/command.py +++ b/lesana/command.py @@ -2,6 +2,12 @@ import logging import os import subprocess +try: + import git + git_available = True +except ImportError: + git_available = False + import gadona from . import Collection, Entry @@ -125,3 +131,63 @@ class Search(gadona.Command): offset, pagesize): print(entry.fname) + + +class Init(gadona.Command): + name = 'init' + arguments = [ + (['--collection', '-c'], dict( + help='The directory to work on (default .)', + default='.' + )), + (['--no-git'], dict( + help='Skip setting up git in this directory', + action="store_false", + dest='git' + )), + ] + + def main(self): + c_dir = self.settings.collection + if self.settings.git: + # Try to initalize a git repo + if git_available: + repo = git.Repo.init(c_dir, bare=False) + else: + log.warning("python3-git not available, could not initalise the git repository.") + repo = None + # Add .lesana directory to .gitignore and add it to the + # staging + lesana_ignored = False + try: + with open(os.path.join(c_dir, '.gitignore'), 'r') as fp: + for line in fp: + print("reading line", line) + if '.lesana' in line: + lesana_ignored = True + continue + except FileNotFoundError: + pass + if not lesana_ignored: + with open(os.path.join(c_dir, '.gitignore'), 'a') as fp: + fp.write('#Added by lesana init\n.lesana') + if repo: + repo.index.add(['.gitignore']) + # TODO: Add hook to index files as they are pulled + # If it doesn't exist, create a skeleton of settings.yaml file + # and then open it for editing + filepath = os.path.join(c_dir, 'settings.yaml') + if not os.path.exists(filepath): + # TODO: write this in a file and just copy that + with open(filepath, 'w') as fp: + fp.write('name: \n') + fp.write('lang: english\n') + fp.write('fields:\n') + try: + subprocess.call(['sensible-editor', filepath]) + except FileNotFoundError as e: + logging.warning( + "Could not open new file with editor: {}".format(str(e)) + ) + if self.settings.git and repo: + repo.index.add(['settings.yaml']) diff --git a/scripts/lesana b/scripts/lesana index a452c64..a500bb8 100755 --- a/scripts/lesana +++ b/scripts/lesana @@ -1,7 +1,7 @@ #!/usr/bin/env python3 import gadona -from lesana.command import New, Edit, Index, Search +from lesana.command import New, Edit, Index, Search, Init if __name__ == '__main__': app = gadona.App() @@ -11,5 +11,6 @@ if __name__ == '__main__': Edit(), Index(), Search(), + Init(), ] app.main() |