diff --git a/mods/db.py b/mods/db.py index 66dd2e6..a12825b 100644 --- a/mods/db.py +++ b/mods/db.py @@ -1,9 +1,14 @@ import os from peewee import Model, CharField, DateTimeField, TextField, ForeignKeyField, DateField, fn from playhouse.sqlite_ext import CSqliteExtDatabase, FTS5Model, AutoIncrementField, SearchField, RowIDField, JSONField -import re +import platform -var_dir = os.path.join(os.path.abspath(os.environ['HOME']), '.eldoc') +if 'windows' in platform.system().lower(): + user_dir = os.path.abspath(os.environ['APPDATA']) +else: + user_dir = os.path.abspath(os.environ['HOME']) + +var_dir = os.path.join(user_dir, '.eldoc') db_dir = os.path.join(var_dir, 'db') db_file = os.path.join(db_dir, 'eldoc.sqlite3')