Onionr/src/utils/createdirs.py

57 lines
1.8 KiB
Python
Raw Normal View History

"""Onionr - Private P2P Communication.
2019-08-06 03:10:21 +00:00
Create required Onionr directories
"""
import os
import stat
from pwd import getpwuid
from getpass import getuser
from . import identifyhome
import filepaths
import onionrexceptions
"""
2019-08-06 03:10:21 +00:00
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
2019-07-18 17:40:48 +00:00
home = identifyhome.identify_home()
def find_owner(filename):
return getpwuid(os.stat(filename).st_uid).pw_name
2019-07-20 06:02:30 +00:00
def create_dirs():
"""Create onionr data-related directories in
order of the hardcoded list below,
2019-09-14 06:29:31 +00:00
then trigger creation of DBs"""
gen_dirs = [home, filepaths.block_data_location,
2020-02-02 09:23:59 +00:00
filepaths.contacts_location, filepaths.export_location]
2019-07-27 02:42:55 +00:00
for path in gen_dirs:
if not os.path.exists(path):
os.makedirs(path)
else:
if getuser() != find_owner(path):
raise onionrexceptions.InsecureDirectoryUsage(
"Directory " + path +
" already exists and is not owned by the same user")
2019-07-18 17:40:48 +00:00
2020-02-04 22:58:24 +00:00
os.chmod(home, stat.S_IRWXU)
from onionrsetup import dbcreator
2019-07-20 06:02:30 +00:00
for db in dbcreator.create_funcs:
try:
db()
except FileExistsError:
pass