Inicializační funkce pro samostatné programy a funkce get_now()

Od nynějška je vždy inicializováno mo.now a je nastavené české locale.

bin/add-role

@@ -4,7 +4,7 @@ import argparse
 
 import mo.db as db
 import mo.util
-from mo.util import die, default_logging
+from mo.util import die, init_standalone
 
 parser = argparse.ArgumentParser(description='Přidělí uživateli roli')
 parser.add_argument('--email', type=str, help='e-mailová adresa uživatele')
@@ -16,7 +16,7 @@ parser.add_argument('--year', type=int, help='omezení role na daný ročník')
 parser.add_argument('--seq', type=int, help='omezení role na dané pořadí kola')
 
 args = parser.parse_args()
-default_logging()
+init_standalone()
 session = db.get_session()
 
 if args.email and args.uid:

bin/create-contests

@@ -4,7 +4,7 @@ import argparse
 
 import mo.db as db
 import mo.util
-from mo.util import die, default_logging
+from mo.util import die, init_standalone
 
 parser = argparse.ArgumentParser(description='Založí soutěže pro dané kolo')
 parser.add_argument(dest='round', type=str, metavar='YY-C-S', help='ID kola')
@@ -12,7 +12,7 @@ parser.add_argument('-n', '--dry-run', default=False, action='store_true', help=
 
 args = parser.parse_args()
 
-default_logging()
+init_standalone()
 sess = db.get_session()
 
 round_code = mo.util.RoundCode.parse(args.round)

bin/create-round

@@ -14,7 +14,7 @@ parser.add_argument('-n', '--name', type=str, required=True, help='název kola')
 
 args = parser.parse_args()
 
-mo.util.default_logging()
+mo.util.init_standalone()
 sess = db.get_session()
 
 rnd = db.Round(

bin/create-user

@@ -16,7 +16,7 @@ parser.add_argument('--mail', default=False, action='store_true', help='pošle u
 
 args = parser.parse_args()
 
-mo.util.default_logging()
+mo.util.init_standalone()
 session = db.get_session()
 
 user = db.User(

bin/init-regions

@@ -6,7 +6,7 @@ import sys
 import mo.db as db
 import mo.util
 
-mo.util.default_logging()
+mo.util.init_standalone()
 session = db.get_session()
 
 session.query(db.Place).delete()

bin/init-schools

@@ -18,7 +18,7 @@ from collections import defaultdict
 import mo.db as db
 import mo.util
 
-mo.util.default_logging()
+mo.util.init_standalone()
 session = db.get_session()
 new_town_cnt = 0
 processed_school_cnt = 0

mo/util.py

@@ -2,8 +2,10 @@
 
 from dataclasses import dataclass
 import datetime
+import dateutil.tz
 import email.message
 import email.headerregistry
+import locale
 import logging
 import re
 import subprocess
@@ -12,6 +14,7 @@ from typing import Any, Optional, NoReturn
 import textwrap
 import urllib.parse
 
+import mo
 import mo.db as db
 import mo.config as config
 
@@ -24,8 +27,15 @@ logger.setLevel(logging.DEBUG)
 logger.propagate = True
 
 
-def default_logging():
-    logging.basicConfig()
+def get_now() -> datetime.datetime:
+    return datetime.datetime.now(tz=dateutil.tz.UTC)
+
+
+def init_standalone():
+    """Společná inicializační funkce pro samostatné programy nezávislé na webu."""
+    logging.basicConfig(format='%(asctime)-15s %(levelname)-5.5s %(message)s')
+    mo.now = get_now()
+    locale.setlocale(locale.LC_COLLATE, 'cs_CZ.UTF-8')
 
 
 def log(type: db.LogType, what: int, details: Any):

mo/web/__init__.py

@@ -14,6 +14,7 @@ import mo
 import mo.config as config
 import mo.db as db
 import mo.users
+import mo.util
 from mo.util import logger
 
 # Aplikační objekt
@@ -62,7 +63,7 @@ def init_request():
     else:
         user = None
     g.user = user
-    mo.now = datetime.datetime.now(tz=dateutil.tz.UTC)
+    mo.now = mo.util.get_now()
     g.now = mo.now  # for templates
     mo.util.current_log_user = user