db: handle database.url_file
This commit is contained in:
		@@ -1,7 +1,16 @@
 | 
				
			|||||||
 | 
					from pathlib import Path
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from sqlalchemy import create_engine
 | 
					from sqlalchemy import create_engine
 | 
				
			||||||
from sqlalchemy.orm import sessionmaker
 | 
					from sqlalchemy.orm import sessionmaker
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from dibbler.conf import config
 | 
					from dibbler.conf import config
 | 
				
			||||||
 | 
					
 | 
				
			||||||
engine = create_engine(config.get("database", "url"))
 | 
					if (url := config.get("database", "url")) is not None:
 | 
				
			||||||
 | 
					  database_url = url
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					elif (url_file := config.get("database", "url_file")) is not None:
 | 
				
			||||||
 | 
					  with Path(url_file).open() as file:
 | 
				
			||||||
 | 
					    database_url = file.read().strip()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					engine = create_engine(database_url)
 | 
				
			||||||
Session = sessionmaker(bind=engine)
 | 
					Session = sessionmaker(bind=engine)
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user