From fce1e73ca5b24a3713b10062462bddab3ae39035 Mon Sep 17 00:00:00 2001
From: Joscha <joscha@plugh.de>
Date: Sun, 14 Apr 2019 19:41:42 +0000
Subject: [PATCH] Allow wtf.py to be imported

---
 wtf.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/wtf.py b/wtf.py
index fb742dc..1e6bde5 100644
--- a/wtf.py
+++ b/wtf.py
@@ -3,7 +3,11 @@ import logging
 import re
 
 import yaboli
-from wtfdb import WtfDB
+
+if __name__ == "__main__":
+    from wtfdb import WtfDB
+else:
+    from .wtfdb import WtfDB
 
 logger = logging.getLogger(__name__)