Fix relative imports to allow running as module.

telegram_stats_bot/log_storage.py

@@ -5,8 +5,8 @@ import os
 
 from sqlalchemy import MetaData, Table, Column, create_engine, BigInteger, TIMESTAMP, Text
 
-from parse import MessageDict
-from db import init_dbs
+from .parse import MessageDict
+from .db import init_dbs
 
 logger = logging.getLogger(__name__)
 metadata = MetaData()

telegram_stats_bot/main.py

@@ -12,9 +12,9 @@ from telegram.ext.dispatcher import run_async
 from telegram.update import Update
 import appdirs
 
-from parse import parse_message
-from log_storage import JSONStore, PostgresStore
-from stats import StatsRunner, get_parser, HelpException
+from .parse import parse_message
+from .log_storage import JSONStore, PostgresStore
+from .stats import StatsRunner, get_parser, HelpException
 
 
 warnings.filterwarnings("ignore")

telegram_stats_bot/stats.py

@@ -12,7 +12,7 @@ import seaborn as sns
 from matplotlib.figure import Figure
 from sqlalchemy.engine import Engine
 
-from utils import escape_markdown
+from .utils import escape_markdown
 
 sns.set_context('paper')
 sns.set_style('whitegrid')