Allow wtf.py to be imported

This commit is contained in:
Joscha 2019-04-14 19:41:42 +00:00
parent b7ccd4f59f
commit fce1e73ca5

6
wtf.py
View file

@ -3,7 +3,11 @@ import logging
import re import re
import yaboli import yaboli
from wtfdb import WtfDB
if __name__ == "__main__":
from wtfdb import WtfDB
else:
from .wtfdb import WtfDB
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)