summaryrefslogtreecommitdiff
path: root/cmount.py
blob: 587c02f591a9af86fa940760ab966f170e202f5a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import json
import os
import subprocess
import sys


def main():
    """
    """
    if (len(sys.argv) != 2):
        sys.exit('Invalid parameter. Usage: {0} <mapping>\n'.format(
            sys.argv[0]))
    cstm_error = ''
    try:
        cstm_error = 'Could not open config (~/.cmount.json)'
        with open(os.path.expanduser('~/.cmount.json')) as fp:
            config_dict = json.load(fp)
        cstm_error = 'Invalid / nonexistent mapping: {0}'.format(sys.argv[1])
        data = config_dict[sys.argv[1]]
        if data.get('type') == 'efile':
            cstm_error = 'Unable to create loop device'
            data['device'] = subprocess.check_output(['/usr/sbin/losetup',
                                                      '--find',
                                                      '--show',
                                                      data['device']]).strip()
        cstm_error = 'system() error'
        subprocess.check_call(['/sbin/cryptsetup',
                               'luksOpen',
                               data['device'],
                               data['mapper_name']])
        arguments = ['/bin/mount', '/dev/mapper/' + data['mapper_name']]
        if data.get('target'):
            arguments.append(data['target'])
        subprocess.check_call(arguments)
    except subprocess.CalledProcessError as e:
        sys.stderr.write('Execution error: {0}\n'.format(e))
        sys.stderr.flush()
    except Exception as e:
        sys.stderr.write('Error: {0}\n'.format(cstm_error))
        sys.stderr.flush()


if __name__ == '__main__':
    main()