-
Notifications
You must be signed in to change notification settings - Fork 0
/
FileOperations.py
44 lines (35 loc) · 1.14 KB
/
FileOperations.py
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
import os
import sys
from pwd import getpwnam
from grp import getgrnam
class Error(Exception):
"""Base class for other exceptions"""
pass
class DirectoryNotExists(Error):
"""Raised when directory doesn't exists"""
pass
class DirectoryPathIsNotAbs(Error):
"""Raised when directory path isn't absolute"""
pass
class DirectoryIsNotAccessible(Error):
"""Raise when directory isn't accessible"""
pass
class FileOperations():
"""Common class for file operations"""
dfile = None
def __init__(self, dfile):
self.dfile = dfile
def setFileOwner(self, user='postgres', group='postgres'):
try:
os.chown(self.dfile, getpwnam(user).pw_uid, getgrnam(group).gr_gid)
except KeyError:
print 'Can\'t access unix user and password database', sys.exc_info()[0]
raise
except OSError:
print 'Can\'t change owner/group for target', sys.exc_info()[0]
raise
def setFilePerm(self, perm=0700):
try:
os.chmod(self.dfile, perm)
except OSError:
print 'Can\'t change permissions for target', sys.exc_info()[0]