basic working config loader
This commit is contained in:
parent
0d0750fa96
commit
53102be98b
|
@ -170,4 +170,7 @@ data/
|
|||
|
||||
# openai-whisper
|
||||
share
|
||||
share/
|
||||
share/
|
||||
|
||||
# local config
|
||||
.config.yaml
|
|
@ -0,0 +1,3 @@
|
|||
username: "username"
|
||||
password: "password"
|
||||
url: "url.org"
|
|
@ -1,12 +1,15 @@
|
|||
#!/usr/bin/env python3
|
||||
from notescriber import config
|
||||
from notescriber import dispatcher
|
||||
from notescriber import notetaker
|
||||
from notescriber import picker
|
||||
from notescriber import transcriber
|
||||
|
||||
config = config.Config()
|
||||
config.load_config(".config.yaml")
|
||||
dispatcher = dispatcher.Dispatcher()
|
||||
notetaker = notetaker.Notetaker()
|
||||
picker = picker.Picker("/home/octopusx/Code/test",dispatcher)
|
||||
notetaker = notetaker.Notetaker(config.username,config.password, config.url)
|
||||
picker = picker.Picker(config.path, dispatcher)
|
||||
transcriber = transcriber.Transcriber()
|
||||
|
||||
dispatcher.add_note_generator(notetaker)
|
||||
|
|
|
@ -0,0 +1,27 @@
|
|||
import argparse
|
||||
import yaml
|
||||
|
||||
class Config:
|
||||
|
||||
username = None
|
||||
password = None
|
||||
url = None
|
||||
path = None
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def parse_args(self):
|
||||
parser = argparse.ArgumentParser(description='Process some integers.')
|
||||
parser.add_argument('--path', type=str, help='Path to config yaml file')
|
||||
args = parser.parse_args()
|
||||
return args.arg
|
||||
|
||||
def load_config(self, path=None):
|
||||
if path is None:
|
||||
path = '.config.yaml'
|
||||
confFile = yaml.safe_load(open(path))
|
||||
self.username = confFile['username']
|
||||
self.password = confFile['password']
|
||||
self.url = confFile['url']
|
||||
self.path = confFile['path']
|
|
@ -4,12 +4,14 @@ class Notetaker:
|
|||
|
||||
username = None
|
||||
password = None
|
||||
url = None
|
||||
|
||||
def __init__(self, username, password):
|
||||
def __init__(self, username, password, url):
|
||||
self.username = username
|
||||
self.password = password
|
||||
self.url = url
|
||||
|
||||
def generate_note(self, filename, note):
|
||||
api = NotesApi(self.username, self.password, 'cloud.octopusx.de')
|
||||
api = NotesApi(self.username, self.password, self.url)
|
||||
ncnote = Note(title=filename, content=note, category="Voicenotes")
|
||||
api.create_note(ncnote)
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
openai-whisper==20231117
|
||||
pyinotify==0.9.6
|
||||
nextcloud_notes_api==1.0.0
|
||||
nextcloud_notes_api==1.0.0
|
||||
PyYaml==6.0.1
|
|
@ -0,0 +1,16 @@
|
|||
import unittest
|
||||
from notescriber.config import Config
|
||||
|
||||
class TranscriptorTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
pass
|
||||
|
||||
def tearDown(self):
|
||||
pass
|
||||
|
||||
def test_run(self):
|
||||
cfg = Config()
|
||||
cfg.load_config(path="example.config.yaml")
|
||||
self.assertEqual("username", cfg.username)
|
||||
self.assertEqual("password", cfg.password)
|
||||
self.assertEqual("url.org", cfg.url)
|
|
@ -9,6 +9,6 @@ class TranscriptorTest(unittest.TestCase):
|
|||
pass
|
||||
|
||||
def test_run(self):
|
||||
ntk = Notetaker()
|
||||
ntk.generate_note("Test Note Title","This is my note")
|
||||
ntk = Notetaker("title","content", "url")
|
||||
# ntk.generate_note("Test Note Title","This is my note")
|
||||
self.assertEqual(True, True)
|
||||
|
|
Loading…
Reference in New Issue