From 39944a56a860836c62b5358174be9e65bd66dc66 Mon Sep 17 00:00:00 2001
From: tassaron
Date: Sun, 28 May 2017 19:08:50 -0400
Subject: create data directory structure

---
 main.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

(limited to 'main.py')

diff --git a/main.py b/main.py
index 7cae950..7ddc4f2 100644
--- a/main.py
+++ b/main.py
@@ -1,11 +1,12 @@
 import sys, io, os
-from PyQt4 import QtCore, QtGui, uic
 from os.path import expanduser
 import atexit
 from queue import Queue
-from PyQt4.QtCore import QSettings, QModelIndex
 import signal
 from importlib import import_module
+from PyQt4 import QtCore, QtGui, uic
+from PyQt4.QtCore import QSettings, QModelIndex
+from PyQt4.QtGui import QDesktopServices
 
 import preview_thread, core, video_thread
 
@@ -123,6 +124,14 @@ class Main(QtCore.QObject):
     self.settings = QSettings('settings.ini', QSettings.IniFormat)
     LoadDefaultSettings(self)
 
+    # create data directory structure if needed
+    dataDir = QDesktopServices.storageLocation(QDesktopServices.DataLocation)
+    if not os.path.exists(dataDir):
+        os.makedirs(dataDir)
+    for neededDirectory in ('projects', 'presets'):
+        if not os.path.exists(os.path.join(dataDir, neededDirectory)):
+            os.mkdir(os.path.join(dataDir, neededDirectory))
+
     self.pages = []
 
     self.previewQueue = Queue()
@@ -382,6 +391,8 @@ else:
 # gui mode
 if __name__ == "__main__":
     app = QtGui.QApplication(sys.argv)
+    app.setApplicationName("audio-visualizer")
+    app.setOrganizationName("audio-visualizer")
     window = uic.loadUi("mainwindow.ui")
     # window.adjustSize()
     desc = QtGui.QDesktopWidget()
-- 
cgit v1.2.3