Add stash init example

- Adds a flag that enables a sort of 'tutorial' mode where an example key is inserted to the stash for the user to try out and `get`

README.md

@@ -74,6 +74,35 @@ Commands:
   ssh      Use a key to SSH-connect to a machine
   unlock   Unlock a key
 
+----- (Optional) -----
+
+# Initializing a stash with a tutorial
+$ ghost init --tutorial
+Stash: tinydb at ~/.ghost/ghost.json
+Initializing stash...
+Initialized stash at: ~/.ghost/ghost.json
+Your passphrase can be found under the `passphrase.ghost` file in the current directory.
+Make sure you save your passphrase somewhere safe. If lost, you will lose access to your stash.
+TUTORIAL: 
+For your convenience we've stored an example key named 'example'. Usually this is done with 'ghost put example first_key=first_value [second_key=second_value [...]]'. You may retrieve it with 'ghost get example'. You may also delete it with 'ghost delete example'.
+
+$ export GHOST_PASSPHRASE=$(cat passphrase.ghost)
+
+$ ghost get example
+Stash: tinydb at ~/.ghost/ghost.json
+Retrieving key...
+
+Description:   This is the key description
+Lock:          False
+Created_At:    2017-05-29 13:49:56
+Modified_At:   2017-05-29 13:49:56
+Value:         key=value;
+Uid:           f7e236f7-450c-400b-9328-95e18d5fac52
+Metadata:      some_key=some_value;
+Type:          secret
+Name:          example
+
+----- (Optional) -----
 
 # Initializing a stash
 $ ghost init

ghost.py

@@ -103,6 +103,13 @@
 
 PASSPHRASE_FILENAME = 'passphrase.ghost'
 
+EXAMPLE_KEY = {
+    'name': 'example',
+    'value': {'key': 'value'},
+    'metadata': {'some_key': 'some_value'},
+    'description': 'This is the key description'
+}
+
 POTENTIAL_PASSPHRASE_LOCATIONS = [
     os.path.abspath(PASSPHRASE_FILENAME),
     os.path.join(GHOST_HOME, PASSPHRASE_FILENAME),
@@ -1137,7 +1144,6 @@ def main():
 
 
 @main.command(name='init', short_help='Initialize a stash')
-@click.argument('STASH_PATH', required=False, type=click.STRING)
 @click.option('-p',
               '--passphrase',
               default=None,
@@ -1149,7 +1155,14 @@ def main():
               default='tinydb',
               type=click.Choice(STORAGE_MAPPING.keys()),
               help='Storage backend for the stash')
-def init_stash(stash_path, passphrase, passphrase_size, backend):
+@click.option('-t',
+              '--tutorial',
+              is_flag=True,
+              help='Whether to create example key')
+@stash_option
+@passphrase_option
+@backend_option
+def init_stash(stash, passphrase, passphrase_size, backend, tutorial):
     r"""Init a stash
 
     `STASH_PATH` is the path to the storage endpoint. If this isn't supplied,
@@ -1167,7 +1180,7 @@ def init_stash(stash_path, passphrase, passphrase_size, backend):
 
     export GHOST_BACKEND='tinydb'
     """
-    stash_path = stash_path or STORAGE_DEFAULT_PATH_MAPPING[backend]
+    stash_path = stash or STORAGE_DEFAULT_PATH_MAPPING[backend]
     click.echo('Stash: {0} at {1}'.format(backend, stash_path))
     storage = STORAGE_MAPPING[backend](**_parse_path_string(stash_path))
 
@@ -1212,6 +1225,14 @@ def init_stash(stash_path, passphrase, passphrase_size, backend):
         'Make sure you save your passphrase somewhere safe. '
         'If lost, you will lose access to your stash.')
 
+    if tutorial:
+        stash.put(**EXAMPLE_KEY)
+        click.echo("TUTORIAL: \nFor your convenience we've stored an example "
+                   "key named 'example'. Usually this is done with 'ghost put "
+                   "example first_key=first_value [second_key=second_value "
+                   "[...]]'. You may retrieve it with 'ghost get example'. You"
+                   " may also delete it with 'ghost delete example'.")
+
 
 @main.command(name='put', short_help='Insert a new key')
 @click.argument('KEY_NAME')

tests/test_ghost.py

@@ -862,6 +862,13 @@ def assert_stash_initialized(stash_path):
     assert len(db) == 1
 
 
+def assert_stash_initialized_tutorial(stash_path):
+    db = get_tinydb(stash_path)
+    assert '2' in db
+    assert db['2']['name'] == 'example'
+    assert len(db) == 2
+
+
 def assert_key_put(db, dont_verify_value=False):
     key = db['2']
     assert key['name'] == 'aws'
@@ -1227,7 +1234,7 @@ def test_cli_stash(stash_path):
     os.close(fd)
     os.remove(passphrase_file_path)
     ghost.PASSPHRASE_FILENAME = passphrase_file_path
-    _invoke('init_stash "{0}"'.format(stash_path))
+    _invoke('init_stash -s "{0}"'.format(stash_path))
     os.environ['GHOST_STASH_PATH'] = stash_path
     with open(passphrase_file_path) as passphrase_file:
         passphrase = passphrase_file.read()
@@ -1271,7 +1278,7 @@ def test_init(self, test_cli_stash):
         assert_stash_initialized(test_cli_stash._storage.db_path)
 
     def test_init_already_initialized(self, test_cli_stash):
-        result = _invoke('init_stash "{0}" -p {1}'.format(
+        result = _invoke('init_stash -s "{0}" -p {1}'.format(
             os.environ['GHOST_STASH_PATH'], test_cli_stash.passphrase))
         assert 'Stash already initialized' in result.output
         assert result.exit_code == 0
@@ -1282,7 +1289,7 @@ def test_init_permission_denied_on_passphrase(self):
         fd, temp_file = tempfile.mkstemp()
         os.close(fd)
         os.remove(temp_file)
-        result = _invoke('init_stash "{0}" -p whatever'.format(temp_file))
+        result = _invoke('init_stash -s "{0}" -p whatever'.format(temp_file))
         assert 'Expected OSError' in str(result.exception)
         assert 'Removing stale stash and passphrase' in str(result.output)
         assert type(result.exception) == SystemExit
@@ -1497,8 +1504,8 @@ def test_load(self, test_cli_stash, temp_file_path):
     def test_fail_init_two_stashes_passphrase_file_exists(self,
                                                           stash_path,
                                                           temp_file_path):
-        _invoke('init_stash "{0}"'.format(stash_path))
-        result = _invoke('init_stash "{0}" -b sqlalchemy'.format(
+        _invoke('init_stash -s "{0}"'.format(stash_path))
+        result = _invoke('init_stash -s "{0}" -b sqlalchemy'.format(
             temp_file_path))
 
         assert 'Overwriting might prevent you' in result.output
@@ -1642,6 +1649,10 @@ def test_ssh_with_key_path_and_extension(self, test_cli_stash):
             'ssh -i /path/to/key [email protected] -o Key="Value"'
         assert expected_command in str(result.exception)
 
+    def test_tutorial(self, stash_path):
+        _invoke('init_stash -s "{0}" -t'.format(stash_path))
+        assert_stash_initialized_tutorial(stash_path)
+
 
 class TestMultiStash:
     # TODO: Test that migrate works when using multi-stash mode