Skip to content
Snippets Groups Projects

Compare revisions

Changes are shown as if the source revision was being merged into the target revision. Learn more about comparing revisions.

Source

Select target project
No results found
Select Git revision
  • devel
  • fo
  • fo-base
  • honza/add-contestant
  • honza/kolo-vs-soutez
  • honza/mr6
  • honza/mr7
  • honza/mra
  • honza/mrd
  • honza/mrf
  • honza/submit-images
  • jh-stress-test-wip
  • jirka/typing
  • jk/issue-196
  • jk/issue-96
  • master
  • mj/submit-images
  • shorten-schools
18 results

Target

Select target project
  • mj/mo-submit
1 result
Select Git revision
  • devel
  • fo
  • fo-base
  • honza/add-contestant
  • honza/kolo-vs-soutez
  • honza/mr6
  • honza/mr7
  • honza/mra
  • honza/mrd
  • honza/mrf
  • honza/submit-images
  • jh-stress-test-wip
  • jirka/typing
  • jk/issue-196
  • jk/issue-96
  • master
  • mj/submit-images
  • shorten-schools
18 results
Show changes
Commits on Source (47)
...@@ -4,6 +4,7 @@ import argparse ...@@ -4,6 +4,7 @@ import argparse
import sys import sys
import mo.db as db import mo.db as db
import mo.email
import mo.users import mo.users
import mo.util import mo.util
...@@ -13,8 +14,7 @@ parser.add_argument(dest='first_name', help='křestní jméno (jedno nebo více) ...@@ -13,8 +14,7 @@ parser.add_argument(dest='first_name', help='křestní jméno (jedno nebo více)
parser.add_argument(dest='last_name', help='příjmení (jedno nebo více)') parser.add_argument(dest='last_name', help='příjmení (jedno nebo více)')
parser.add_argument('--org', default=False, action='store_true', help='přidělí uživateli organizátorská práva') parser.add_argument('--org', default=False, action='store_true', help='přidělí uživateli organizátorská práva')
parser.add_argument('--admin', default=False, action='store_true', help='přidělí uživateli správcovská práva') parser.add_argument('--admin', default=False, action='store_true', help='přidělí uživateli správcovská práva')
parser.add_argument('--passwd', type=str, help='nastaví počáteční heslo') parser.add_argument('--passwd', type=str, help='nastaví počáteční heslo (jinak pošle aktivační mail)')
parser.add_argument('--mail', default=False, action='store_true', help='pošle uživateli mail o založení účtu')
args = parser.parse_args() args = parser.parse_args()
email = mo.users.normalize_email(args.email) email = mo.users.normalize_email(args.email)
...@@ -45,11 +45,9 @@ mo.util.log(db.LogType.user, user.user_id, { ...@@ -45,11 +45,9 @@ mo.util.log(db.LogType.user, user.user_id, {
if args.passwd is not None: if args.passwd is not None:
mo.users.set_password(user, args.passwd) mo.users.set_password(user, args.passwd)
token = mo.users.make_activation_token(user)
if args.mail:
token = mo.users.ask_reset_password(user)
session.commit() session.commit()
if args.mail: if args.passwd is None:
mo.util.send_new_account_email(user, token) mo.email.send_new_account_email(user, token)
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse import argparse
import sys
import mo.config as config import mo.config
import mo.email
import mo.db as db import mo.db as db
import mo.users import mo.users
import mo.util import mo.util
parser = argparse.ArgumentParser(description='Resetuje uživateli heslo a pošle mail') parser = argparse.ArgumentParser(description='Pošle uživateli nový aktivační mail')
parser.add_argument(dest='email', help='e-mailová adresa') parser.add_argument(dest='email', help='e-mailová adresa')
parser.add_argument('--new', default=False, action='store_true', help='pošle mail o založení účtu')
parser.add_argument('--mail-instead', metavar='EMAIL', default=None, help='pošle mail někomu jinému') parser.add_argument('--mail-instead', metavar='EMAIL', default=None, help='pošle mail někomu jinému')
args = parser.parse_args() args = parser.parse_args()
...@@ -22,13 +21,10 @@ user = mo.users.user_by_email(args.email) ...@@ -22,13 +21,10 @@ user = mo.users.user_by_email(args.email)
if user is None: if user is None:
mo.util.die('Tento uživatel neexistuje') mo.util.die('Tento uživatel neexistuje')
token = mo.users.ask_reset_password(user) token = mo.users.make_activation_token(user)
session.commit() session.commit()
if args.mail_instead: if args.mail_instead:
mo.config.MAIL_INSTEAD = args.mail_instead mo.config.MAIL_INSTEAD = args.mail_instead
if args.new: mo.email.send_new_account_email(user, token)
mo.util.send_new_account_email(user, token)
else:
mo.util.send_password_reset_email(user, token)
-- CREATE ROLE mo_osmo LOGIN PASSWORD 'pass'; -- CREATE ROLE mo_osmo LOGIN PASSWORD 'pass';
-- CREATE DATABASE mo_osmo WITH OWNER=mo_osmo; -- CREATE DATABASE mo_osmo WITH OWNER=mo_osmo;
-- GRANT mo_osmo TO some_admin; -- GRANT mo_osmo TO some_admin;
-- CREATE EXTENSION unaccent;
SET ROLE mo_osmo; SET ROLE mo_osmo;
-- Funkce pro odakcentování textu pomocí extension unaccent.
-- Je immutable, takže se dá používat i v indexech.
-- Zdroj: http://stackoverflow.com/questions/11005036/does-postgresql-support-accent-insensitive-collations
CREATE OR REPLACE FUNCTION f_unaccent(text)
RETURNS text AS
$func$
SELECT unaccent('unaccent', $1)
$func$ LANGUAGE sql IMMUTABLE SET search_path = public, pg_temp;
-- Uživatelský účet -- Uživatelský účet
CREATE TABLE users ( CREATE TABLE users (
user_id serial PRIMARY KEY, user_id serial PRIMARY KEY,
...@@ -15,7 +25,7 @@ CREATE TABLE users ( ...@@ -15,7 +25,7 @@ CREATE TABLE users (
is_test boolean NOT NULL DEFAULT false, -- testovací účastník, není vidět ve výsledkovkách is_test boolean NOT NULL DEFAULT false, -- testovací účastník, není vidět ve výsledkovkách
created_at timestamp with time zone NOT NULL DEFAULT CURRENT_TIMESTAMP, created_at timestamp with time zone NOT NULL DEFAULT CURRENT_TIMESTAMP,
last_login_at timestamp with time zone DEFAULT NULL, last_login_at timestamp with time zone DEFAULT NULL,
reset_at timestamp with time zone DEFAULT NULL, -- poslední požadavek na reset hesla reset_at timestamp with time zone DEFAULT NULL, -- poslední reset/aktivace nebo žádost o ně
password_hash varchar(255) DEFAULT NULL, -- heš hesla (je-li nastaveno) password_hash varchar(255) DEFAULT NULL, -- heš hesla (je-li nastaveno)
note text NOT NULL DEFAULT '' -- poznámka viditelná pro orgy note text NOT NULL DEFAULT '' -- poznámka viditelná pro orgy
); );
...@@ -45,6 +55,8 @@ CREATE TABLE places ( ...@@ -45,6 +55,8 @@ CREATE TABLE places (
); );
CREATE INDEX places_parent_index ON places (parent); CREATE INDEX places_parent_index ON places (parent);
-- XXX: Potřebujeme operator class text_pattern_ops, aby index fungoval na prefixové LIKE
CREATE INDEX places_noacc_index ON places ((lower(f_unaccent(name))) text_pattern_ops);
-- Rekurzivní dotaz na nadřazené regiony: -- Rekurzivní dotaz na nadřazené regiony:
-- WITH RECURSIVE parent_regions(parent, place_id) AS ( -- WITH RECURSIVE parent_regions(parent, place_id) AS (
...@@ -89,6 +101,12 @@ CREATE TYPE score_mode AS ENUM ( ...@@ -89,6 +101,12 @@ CREATE TYPE score_mode AS ENUM (
'mo' -- jednoznačné pořadí podle pravidel MO 'mo' -- jednoznačné pořadí podle pravidel MO
); );
CREATE TYPE enroll_mode AS ENUM ( -- režim přihlašování účastníků
'manual', -- přihlašuje organizátor
'register', -- účastník se registruje
'confirm' -- účastník se registruje, ale org ho musí potvrdit
);
CREATE TABLE rounds ( CREATE TABLE rounds (
round_id serial PRIMARY KEY, round_id serial PRIMARY KEY,
master_round_id int DEFAULT NULL REFERENCES rounds(round_id), master_round_id int DEFAULT NULL REFERENCES rounds(round_id),
...@@ -109,6 +127,8 @@ CREATE TABLE rounds ( ...@@ -109,6 +127,8 @@ CREATE TABLE rounds (
score_successful_limit int DEFAULT NULL, -- bodový limit na označení za úspěšného řešitele score_successful_limit int DEFAULT NULL, -- bodový limit na označení za úspěšného řešitele
points_step numeric(2,1) NOT NULL DEFAULT 1, -- s jakou přesností jsou přidělovány body (celé aneb 1, 0.5, 0.1) points_step numeric(2,1) NOT NULL DEFAULT 1, -- s jakou přesností jsou přidělovány body (celé aneb 1, 0.5, 0.1)
has_messages boolean NOT NULL DEFAULT false, -- má zprávičky has_messages boolean NOT NULL DEFAULT false, -- má zprávičky
enroll_mode enroll_mode NOT NULL DEFAULT 'manual', -- režim přihlašování (pro vyšší kola vždy 'manual')
enroll_advert varchar(255) NOT NULL DEFAULT '', -- popis v přihlašovacím formuláři
UNIQUE (year, category, seq, part) UNIQUE (year, category, seq, part)
); );
...@@ -133,17 +153,19 @@ CREATE TABLE participants ( ...@@ -133,17 +153,19 @@ CREATE TABLE participants (
school int NOT NULL REFERENCES places(place_id), school int NOT NULL REFERENCES places(place_id),
birth_year int NOT NULL, birth_year int NOT NULL,
grade varchar(20) NOT NULL, -- třída ve tvaru "X/Y" grade varchar(20) NOT NULL, -- třída ve tvaru "X/Y"
registered_on timestamp with time zone DEFAULT NULL, -- kdy se účastník přihlásil (NULL, pokud ho přihlásil organizátor)
PRIMARY KEY (user_id, year) PRIMARY KEY (user_id, year)
); );
-- Účast v soutěžním kole -- Účast v soutěžním kole
CREATE TYPE part_state AS ENUM ( CREATE TYPE part_state AS ENUM (
'registered', -- sám se přihlásil 'registered', -- sám se přihlásil, čeká na potvrzení organizátorem
'invited', -- pozván -- 'invited', -- pozván (už nepoužíváme)
'refused', -- odmítl účast 'active', -- soutěží (přihlášku zadal/potvrdil organizátor)
'present', -- soutěžil 'refused', -- organizátor odmítl přihlášku
'absent', -- bez omluvy nedorazil -- 'present', -- soutěžil (už nepoužíváme)
'absent', -- nedorazil
'disqualified' -- diskvalifikovaný 'disqualified' -- diskvalifikovaný
); );
...@@ -312,3 +334,24 @@ CREATE TABLE messages ( ...@@ -312,3 +334,24 @@ CREATE TABLE messages (
markdown text NOT NULL, markdown text NOT NULL,
html text NOT NULL html text NOT NULL
); );
-- Požadavky na registraci a změny vlastností účtu
CREATE TYPE reg_req_type AS ENUM (
'register',
'change_email',
'reset_password'
);
CREATE TABLE reg_requests (
reg_id serial PRIMARY KEY,
type reg_req_type NOT NULL,
created_at timestamp with time zone NOT NULL,
expires_at timestamp with time zone NOT NULL,
used_at timestamp with time zone DEFAULT NULL,
email varchar(255) DEFAULT NULL, -- adresa, kterou potvrzujeme
captcha_token varchar(255) DEFAULT NULL, -- token pro fázi 1 registrace
email_token varchar(255) UNIQUE NOT NULL, -- token pro fázi 2 registrace
user_id int DEFAULT NULL REFERENCES users(user_id) ON DELETE CASCADE,
client varchar(255) NOT NULL -- kdo si registraci vyžádal
);
SET ROLE 'mo_osmo';
CREATE TYPE enroll_mode AS ENUM ( -- režim přihlašování účastníků
'manual', -- přihlašuje organizátor
'register', -- účastník se registruje
'confirm' -- účastník se registruje, ale org ho musí potvrdit
);
ALTER TABLE rounds ADD COLUMN enroll_mode enroll_mode NOT NULL DEFAULT 'manual';
ALTER TABLE rounds ADD COLUMN enroll_advert varchar(255) NOT NULL DEFAULT '';
ALTER TYPE part_state ADD VALUE 'active' AFTER 'invited';
ALTER TABLE participants ADD COLUMN registered_on timestamp with time zone DEFAULT NULL;
UPDATE participations SET state='active' WHERE state IN ('registered', 'invited');
CREATE TYPE reg_req_type AS ENUM (
'register',
'change_email',
'reset_passwd'
);
CREATE TABLE reg_requests (
reg_id serial PRIMARY KEY,
type reg_req_type NOT NULL,
created_at timestamp with time zone NOT NULL,
expires_at timestamp with time zone NOT NULL,
used_at timestamp with time zone DEFAULT NULL,
email varchar(255) DEFAULT NULL,
captcha_token varchar(255) DEFAULT NULL,
email_token varchar(255) UNIQUE NOT NULL,
user_id int DEFAULT NULL REFERENCES users(user_id) ON DELETE CASCADE,
client varchar(255) NOT NULL
);
SET ROLE 'mo_osmo';
-- Musí udělat správce (ale uvnitř naší DB):
-- CREATE EXTENSION unaccent;
CREATE OR REPLACE FUNCTION f_unaccent(text)
RETURNS text AS
$func$
SELECT unaccent('unaccent', $1)
$func$ LANGUAGE sql IMMUTABLE SET search_path = public, pg_temp;
CREATE INDEX places_noacc_index ON places ((lower(f_unaccent(name))) text_pattern_ops);
...@@ -42,12 +42,14 @@ MAX_BATCH_CONTENT_LENGTH = 1000000000 ...@@ -42,12 +42,14 @@ MAX_BATCH_CONTENT_LENGTH = 1000000000
# Adresář, do kterého ukládáme data (pro vývoj relativní, pro instalaci absolutní) # Adresář, do kterého ukládáme data (pro vývoj relativní, pro instalaci absolutní)
DATA_DIR = 'data' DATA_DIR = 'data'
# Jak často se má provádět periodická kontrola dávek [s] # Jak často se má spouštět garbage collector na dávky a tokeny [s]
JOB_GC_PERIOD = 60 GC_PERIOD = 60
# Za jak dlouho expiruje dokončená dávka [min] # Za jak dlouho expiruje dokončená dávka [min]
JOB_EXPIRATION = 5 JOB_EXPIRATION = 5
# Automatické přihlašování účastníků do testovací soutěže # Kolik nejvýše dovolujeme registrací za minutu
# (kolo aktuální_ročník-T-1, celostátní soutěž) REG_MAX_PER_MINUTE = 10
AUTO_REGISTER_TEST = False
# Jak dlouho vydrží tokeny používané při registraci a změnách e-mailu [min]
REG_TOKEN_VALIDITY = 10
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
import datetime import datetime
# Aktuální ročník # Aktuální ročník
current_year = 70 current_year = 71
# Referenční čas nastavovaný v initu requestu (web) nebo při volání skriptu # Referenční čas nastavovaný v initu requestu (web) nebo při volání skriptu
now: datetime.datetime now: datetime.datetime
......
...@@ -16,12 +16,13 @@ from sqlalchemy.orm.attributes import get_history ...@@ -16,12 +16,13 @@ from sqlalchemy.orm.attributes import get_history
from sqlalchemy.dialects.postgresql import JSONB from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql.expression import CTE from sqlalchemy.sql.expression import CTE
from sqlalchemy.sql.functions import ReturnTypeFromArgs
from sqlalchemy.sql.sqltypes import Numeric from sqlalchemy.sql.sqltypes import Numeric
from typing import Optional, List, Tuple from typing import Optional, List, Tuple
import mo import mo
from mo.place_level import place_levels, PlaceLevel from mo.place_level import place_levels, PlaceLevel
from mo.util_format import timedelta, time_and_timedelta from mo.util_format import timeformat_short, timedelta, time_and_timedelta
# HACK: Work-around for https://github.com/dropbox/sqlalchemy-stubs/issues/114 # HACK: Work-around for https://github.com/dropbox/sqlalchemy-stubs/issues/114
from typing import TYPE_CHECKING, TypeVar, Type, Any from typing import TYPE_CHECKING, TypeVar, Type, Any
...@@ -34,6 +35,11 @@ else: ...@@ -34,6 +35,11 @@ else:
from sqlalchemy import Enum from sqlalchemy import Enum
# Funkce f_unaccent je definovaná v db.ddl, naučme Alchymii volat ji:
class f_unaccent(ReturnTypeFromArgs):
pass
Base = declarative_base() Base = declarative_base()
metadata = Base.metadata metadata = Base.metadata
...@@ -99,6 +105,7 @@ class Place(Base): ...@@ -99,6 +105,7 @@ class Place(Base):
nuts = Column(String(255), unique=True, server_default=text("NULL::character varying")) nuts = Column(String(255), unique=True, server_default=text("NULL::character varying"))
note = Column(Text, nullable=False, server_default=text("''::text")) note = Column(Text, nullable=False, server_default=text("''::text"))
parent_place = relationship('Place', primaryjoin='Place.parent == Place.place_id', remote_side='Place.place_id')
children = relationship('Place') children = relationship('Place')
school = relationship('School', uselist=False, back_populates='place') school = relationship('School', uselist=False, back_populates='place')
...@@ -203,6 +210,22 @@ round_score_mode_names = { ...@@ -203,6 +210,22 @@ round_score_mode_names = {
} }
class RoundEnrollMode(MOEnum):
manual = auto()
register = auto()
confirm = auto()
def friendly_name(self) -> str:
return round_enroll_mode_names[self]
round_enroll_mode_names = {
RoundEnrollMode.manual: "Jen organizátoři",
RoundEnrollMode.register: "Účastníci sami",
RoundEnrollMode.confirm: "Potvrzení organizátorem",
}
# V DB jako numeric(2,1), používá se tak snadněji, než enum # V DB jako numeric(2,1), používá se tak snadněji, než enum
round_points_step_names = { round_points_step_names = {
decimal.Decimal('1'): "Celé body", decimal.Decimal('1'): "Celé body",
...@@ -237,6 +260,8 @@ class Round(Base): ...@@ -237,6 +260,8 @@ class Round(Base):
score_successful_limit = Column(Numeric) score_successful_limit = Column(Numeric)
points_step = Column(Numeric, nullable=False) points_step = Column(Numeric, nullable=False)
has_messages = Column(Boolean, nullable=False, server_default=text("false")) has_messages = Column(Boolean, nullable=False, server_default=text("false"))
enroll_mode = Column(Enum(RoundEnrollMode, name='enroll_mode'), nullable=False, server_default=text("'basic'::enroll_mode"))
enroll_advert = Column(String(255), nullable=False, server_default=text("''::text"))
master = relationship('Round', primaryjoin='Round.master_round_id == Round.round_id', remote_side='Round.round_id', post_update=True) master = relationship('Round', primaryjoin='Round.master_round_id == Round.round_id', remote_side='Round.round_id', post_update=True)
...@@ -281,6 +306,14 @@ class Round(Base): ...@@ -281,6 +306,14 @@ class Round(Base):
return round_points_step_names[self.points_step] return round_points_step_names[self.points_step]
return str(self.points_step) return str(self.points_step)
def format_times(self) -> str:
times = []
if self.ct_tasks_start is not None:
times.append('od ' + timeformat_short(self.ct_tasks_start))
if self.ct_submit_end is not None:
times.append('do ' + timeformat_short(self.ct_submit_end))
return " ".join(times)
class User(Base): class User(Base):
__tablename__ = 'users' __tablename__ = 'users'
...@@ -403,6 +436,7 @@ class Participant(Base): ...@@ -403,6 +436,7 @@ class Participant(Base):
school = Column(Integer, ForeignKey('places.place_id'), nullable=False) school = Column(Integer, ForeignKey('places.place_id'), nullable=False)
birth_year = Column(Integer, nullable=False) birth_year = Column(Integer, nullable=False)
grade = Column(String(20), nullable=False) grade = Column(String(20), nullable=False)
registered_on = Column(DateTime(True))
user = relationship('User') user = relationship('User')
school_place = relationship('Place', primaryjoin='Participant.school == Place.place_id') school_place = relationship('Place', primaryjoin='Participant.school == Place.place_id')
...@@ -410,9 +444,8 @@ class Participant(Base): ...@@ -410,9 +444,8 @@ class Participant(Base):
class PartState(MOEnum): class PartState(MOEnum):
registered = auto() registered = auto()
invited = auto() active = auto()
refused = auto() refused = auto()
present = auto()
absent = auto() absent = auto()
disqualified = auto() disqualified = auto()
...@@ -422,9 +455,8 @@ class PartState(MOEnum): ...@@ -422,9 +455,8 @@ class PartState(MOEnum):
part_state_names = { part_state_names = {
PartState.registered: 'přihlášený', PartState.registered: 'přihlášený',
PartState.invited: 'pozvaný', PartState.active: 'soutěží',
PartState.refused: 'odmítnutý', PartState.refused: 'odmítnutý',
PartState.present: 'přítomný',
PartState.absent: 'nepřítomný', PartState.absent: 'nepřítomný',
PartState.disqualified: 'diskvalifikovaný', PartState.disqualified: 'diskvalifikovaný',
} }
...@@ -510,6 +542,12 @@ class UserRole(Base): ...@@ -510,6 +542,12 @@ class UserRole(Base):
return " ".join(parts) return " ".join(parts)
def applies_to(self, at: Optional[Place] = None, year: Optional[int] = None, cat: Optional[str] = None, seq: Optional[int] = None) -> bool:
return ((at is None or self.place_id == at.place_id)
and (self.year is None or year is None or self.year == year)
and (self.category is None or cat is None or self.category == cat or (self.category == 'Z' and cat.startswith('Z')))
and (self.seq is None or seq is None or self.seq == seq))
class PaperType(MOEnum): class PaperType(MOEnum):
solution = auto() solution = auto()
...@@ -644,6 +682,29 @@ class Message(Base): ...@@ -644,6 +682,29 @@ class Message(Base):
created_by_user = relationship('User') created_by_user = relationship('User')
class RegReqType(MOEnum):
register = auto()
change_email = auto()
reset_passwd = auto()
class RegRequest(Base):
__tablename__ = 'reg_requests'
reg_id = Column(Integer, primary_key=True, server_default=text("nextval('reg_requests_reg_id_seq'::regclass)"))
type = Column(Enum(RegReqType, name='reg_req_type'), nullable=False)
created_at = Column(DateTime(True), nullable=False)
expires_at = Column(DateTime(True), nullable=False)
used_at = Column(DateTime(True))
captcha_token = Column(Text)
email = Column(Text)
email_token = Column(Text, nullable=False, unique=True)
user_id = Column(Integer, ForeignKey('users.user_id'))
client = Column(Text, nullable=False)
user = relationship('User')
_engine: Optional[Engine] = None _engine: Optional[Engine] = None
_session: Optional[Session] = None _session: Optional[Session] = None
flask_db: Any = None flask_db: Any = None
......
# Rozesílání e-mailových notifikací všeho druhu
import datetime
import email.message
import email.headerregistry
import subprocess
import textwrap
import urllib.parse
import mo.db as db
import mo.config as config
from mo.util import logger
def send_user_email(user: db.User, subject: str, body: str) -> bool:
logger.info(f'Mail: "{subject}" -> {user.email}')
mail_from = getattr(config, 'MAIL_FROM', None)
if mail_from is None:
logger.error('Mail: V configu chybí nastavení MAIL_FROM')
return False
msg = email.message.EmailMessage()
msg['From'] = email.headerregistry.Address(
display_name='Odevzdávací Systém MO',
addr_spec=mail_from,
)
msg['To'] = [
email.headerregistry.Address(
display_name=user.full_name(),
addr_spec=user.email,
)
]
msg['Reply-To'] = email.headerregistry.Address(
display_name='Správce OSMO',
addr_spec=config.MAIL_CONTACT,
)
msg['Subject'] = 'OSMO – ' + subject
msg['Date'] = datetime.datetime.now()
msg.set_content(body, cte='quoted-printable')
mail_instead = getattr(config, 'MAIL_INSTEAD', None)
if mail_instead is None:
send_to = user.email
else:
send_to = mail_instead
sm = subprocess.Popen(
[
'/usr/sbin/sendmail',
'-oi',
'-f',
mail_from,
send_to,
],
stdin=subprocess.PIPE,
)
sm.communicate(msg.as_bytes())
if sm.returncode != 0:
logger.error('Mail: Sendmail failed with return code {}'.format(sm.returncode))
return False
return True
def activate_url(token: str) -> str:
return config.WEB_ROOT + 'acct/activate?' + urllib.parse.urlencode({'token': token}, safe=':')
def confirm_url(type: str, token: str) -> str:
return config.WEB_ROOT + f'acct/confirm/{type}?' + urllib.parse.urlencode({'token': token}, safe=':')
def contestant_list_url(contest: db.Contest, registered_only: bool) -> str:
url = config.WEB_ROOT + f'org/contest/c/{contest.contest_id}/ucastnici'
if registered_only:
url += '?participation_state=registered'
return url
def send_new_account_email(user: db.User, token: str) -> bool:
return send_user_email(user, 'Založen nový účet', textwrap.dedent('''\
Vítejte!
Právě Vám byl založen účet v Odevzdávacím systému Matematické olympiády.
Nastavte si prosím heslo na následující stránce:
{}
Váš OSMO
'''.format(activate_url(token))))
def send_password_reset_email(user: db.User, token: str) -> bool:
return send_user_email(user, 'Obnova hesla', textwrap.dedent('''\
Někdo (pravděpodobně Vy) požádal o obnovení hesla k Vašemu účtu v Odevzdávacím
systému Matematické olympiády. Heslo si můžete nastavit, případně požadavek
zrušit, na následující stránce:
{}
Váš OSMO
'''.format(confirm_url('p', token))))
def send_confirm_create_email(user: db.User, token: str) -> bool:
return send_user_email(user, 'Založení účtu', textwrap.dedent('''\
Někdo (pravděpodobně Vy) požádal o založení účtu s touto e-mailovou adresou
v Odevzdávacím systému Matematické olympiády. Pokud účet chcete založit,
následujte tento odkaz:
{}
Váš OSMO
'''.format(confirm_url('r', token))))
def send_confirm_change_email(user: db.User, token: str) -> bool:
return send_user_email(user, 'Změna e-mailové adresy', textwrap.dedent('''\
Někdo (pravděpodobně Vy) požádal o nastavení e-mailové adresy k účtu
v Odevzdávacím systému Matematické olympiády na tuto adresu.
Pokud změnu chcete provést, následujte tento odkaz:
{}
Váš OSMO
'''.format(confirm_url('e', token))))
def send_join_notify_email(dest: db.User, who: db.User, contest: db.Contest) -> bool:
round = contest.round
place = contest.place
if contest.round.enroll_mode == db.RoundEnrollMode.confirm:
confirm = 'Přihlášku je potřeba potvrdit v seznamu účastníků soutěže.'
url = 'Přihlášky k potvrzení: ' + contestant_list_url(contest, True)
else:
confirm = 'Přihláška byla schválena automaticky.'
url = 'Seznam účastníků: ' + contestant_list_url(contest, False)
return send_user_email(dest, f'Nový účastník kategorie {round.category}', textwrap.dedent(f'''\
Nový účastník se přihlásil do MO v oblasti, kterou garantujete.
Jméno: {who.full_name()}
E-mail: {who.email}
Kategorie: {round.category}
Kolo: {round.name}
Místo: {place.name}
{confirm}
{url}
Váš OSMO
'''))
...@@ -10,6 +10,7 @@ from typing import List, Optional, Any, Dict, Type, Union ...@@ -10,6 +10,7 @@ from typing import List, Optional, Any, Dict, Type, Union
import mo.csv import mo.csv
from mo.csv import FileFormat, MissingHeaderError from mo.csv import FileFormat, MissingHeaderError
import mo.db as db import mo.db as db
import mo.email
import mo.rights import mo.rights
import mo.users import mo.users
import mo.util import mo.util
...@@ -187,7 +188,7 @@ class Import: ...@@ -187,7 +188,7 @@ class Import:
return r return r
def find_or_create_user(self, email: str, krestni: str, prijmeni: str, is_org: bool) -> Optional[db.User]: def find_or_create_user(self, email: str, krestni: Optional[str], prijmeni: Optional[str], is_org: bool) -> Optional[db.User]:
try: try:
user, is_new = mo.users.find_or_create_user(email, krestni, prijmeni, is_org, reason='import') user, is_new = mo.users.find_or_create_user(email, krestni, prijmeni, is_org, reason='import')
except mo.CheckError as e: except mo.CheckError as e:
...@@ -211,7 +212,7 @@ class Import: ...@@ -211,7 +212,7 @@ class Import:
return pts return pts
def find_or_create_participant(self, user: db.User, year: int, school_id: int, birth_year: int, grade: str) -> Optional[db.Participant]: def find_or_create_participant(self, user: db.User, year: int, school_id: Optional[int], birth_year: Optional[int], grade: Optional[str]) -> Optional[db.Participant]:
try: try:
part, is_new = mo.users.find_or_create_participant(user, year, school_id, birth_year, grade, reason='import') part, is_new = mo.users.find_or_create_participant(user, year, school_id, birth_year, grade, reason='import')
except mo.CheckError as e: except mo.CheckError as e:
...@@ -372,9 +373,9 @@ class Import: ...@@ -372,9 +373,9 @@ class Import:
for uid in self.new_user_ids: for uid in self.new_user_ids:
u = sess.query(db.User).get(uid) u = sess.query(db.User).get(uid)
if u and not u.password_hash and not u.reset_at: if u and not u.password_hash and not u.reset_at:
token = mo.users.ask_reset_password(u) token = mo.users.make_activation_token(u)
sess.commit() sess.commit()
mo.util.send_new_account_email(u, token) mo.email.send_new_account_email(u, token)
else: else:
sess.rollback() sess.rollback()
...@@ -432,28 +433,23 @@ class ContestImport(Import): ...@@ -432,28 +433,23 @@ class ContestImport(Import):
assert isinstance(r, ContestImportRow) assert isinstance(r, ContestImportRow)
num_prev_errs = len(self.errors) num_prev_errs = len(self.errors)
email = self.parse_email(r.email) email = self.parse_email(r.email)
krestni = self.parse_name(r.krestni) krestni = self.parse_name(r.krestni) if r.krestni else None
prijmeni = self.parse_name(r.prijmeni) prijmeni = self.parse_name(r.prijmeni) if r.prijmeni else None
school_place = self.parse_school(r.kod_skoly) school_place = self.parse_school(r.kod_skoly) if r.kod_skoly else None
rocnik = self.parse_grade(r.rocnik, (school_place.school if school_place else None)) rocnik = self.parse_grade(r.rocnik, (school_place.school if school_place else None)) if r.rocnik else None
rok_naroz = self.parse_born(r.rok_naroz) rok_naroz = self.parse_born(r.rok_naroz) if r.rok_naroz else None
misto = self.parse_opt_place(r.kod_mista, 'místo') misto = self.parse_opt_place(r.kod_mista, 'místo')
oblast = self.parse_opt_place(r.kod_oblasti, 'oblast') oblast = self.parse_opt_place(r.kod_oblasti, 'oblast')
if (len(self.errors) > num_prev_errs if (len(self.errors) > num_prev_errs
or email is None or email is None):
or krestni is None
or prijmeni is None
or school_place is None
or rocnik is None
or rok_naroz is None):
return return
user = self.find_or_create_user(email, krestni, prijmeni, is_org=False) user = self.find_or_create_user(email, krestni, prijmeni, is_org=False)
if user is None: if user is None:
return return
part = self.find_or_create_participant(user, mo.current_year, school_place.place_id, rok_naroz, rocnik) part = self.find_or_create_participant(user, mo.current_year, school_place.place_id if school_place else None, rok_naroz, rocnik)
if part is None: if part is None:
return return
......
...@@ -167,7 +167,7 @@ class Rights: ...@@ -167,7 +167,7 @@ class Rights:
return right in self.rights return right in self.rights
def get_roles(self) -> List[db.RoleType]: def get_roles(self) -> List[db.RoleType]:
"""Seznam unikátních rolí, stříděných od nejvýznamnější.""" """Seznam unikátních rolí, setříděných od nejvýznamnější."""
role_set = set(ur.role for ur in self.user_roles) role_set = set(ur.role for ur in self.user_roles)
return sorted(role_set, key=lambda r: role_order_by_type[r]) return sorted(role_set, key=lambda r: role_order_by_type[r])
...@@ -325,10 +325,7 @@ class Gatekeeper: ...@@ -325,10 +325,7 @@ class Gatekeeper:
rights.rights = set() rights.rights = set()
def try_role(role: db.UserRole, at: Optional[db.Place]): def try_role(role: db.UserRole, at: Optional[db.Place]):
if ((at is None or role.place_id == at.place_id) if (role.applies_to(at=at, year=year, cat=cat, seq=seq)
and (role.year is None or year is None or role.year == year)
and (role.category is None or cat is None or role.category == cat or (role.category == 'Z' and cat.startswith('Z')))
and (role.seq is None or seq is None or role.seq == seq)
and (min_role is None or role_order_by_type[min_role] >= role_order_by_type[role.role])): and (min_role is None or role_order_by_type[min_role] >= role_order_by_type[role.role])):
rights.user_roles.append(role) rights.user_roles.append(role)
r = roles_by_type[role.role] r = roles_by_type[role.role]
......
...@@ -106,7 +106,7 @@ class Score: ...@@ -106,7 +106,7 @@ class Score:
def __init__( def __init__(
self, round: db.Round, contest: Optional[db.Contest] = None, self, round: db.Round, contest: Optional[db.Contest] = None,
# Ze kterých stavů chceme výsledkovku počítat # Ze kterých stavů chceme výsledkovku počítat
part_states: List[db.PartState] = [db.PartState.registered, db.PartState.invited, db.PartState.present], part_states: List[db.PartState] = [db.PartState.registered, db.PartState.active],
): ):
self.round = round self.round = round
self.contest = contest self.contest = contest
......
# Podepsané tokeny # Podepsané tokeny
import hashlib
import hmac import hmac
import urllib.parse import urllib.parse
from typing import Optional, List from typing import Optional, List
...@@ -8,9 +9,7 @@ import mo.config as config ...@@ -8,9 +9,7 @@ import mo.config as config
def _sign_token(token: str, use: str) -> str: def _sign_token(token: str, use: str) -> str:
key = '-'.join(('sign-token', use, config.SECRET_KEY)) return sign(token, 'token-' + use)
mac = hmac.HMAC(key.encode('us-ascii'), token.encode('us-ascii'), 'sha256')
return mac.hexdigest()[:16]
def sign_token(fields: List[str], use: str) -> str: def sign_token(fields: List[str], use: str) -> str:
...@@ -27,3 +26,18 @@ def verify_token(token: str, use: str) -> Optional[List[str]]: ...@@ -27,3 +26,18 @@ def verify_token(token: str, use: str) -> Optional[List[str]]:
if _sign_token(':'.join(enc_fields), use) != sign: if _sign_token(':'.join(enc_fields), use) != sign:
return None return None
return [urllib.parse.unquote(f) for f in enc_fields] return [urllib.parse.unquote(f) for f in enc_fields]
def sign(msg: str, use: str) -> str:
"""Podpis parametrizovaný tajným klíčem a účelem."""
key = use + '#' + config.SECRET_KEY
mac = hmac.HMAC(key.encode('us-ascii'), msg.encode('us-ascii'), 'sha256')
return mac.hexdigest()[:16]
def hash(msg: str, use: str) -> str:
"""Hešovací funkce parametrizovaná tajným klíčem a účelem."""
m = '#'.join((use, config.SECRET_KEY, msg)).encode('us-ascii')
return hashlib.sha256(m).hexdigest()
...@@ -2,12 +2,15 @@ ...@@ -2,12 +2,15 @@
import bcrypt import bcrypt
import datetime import datetime
import dateutil.tz
import email.errors import email.errors
import email.headerregistry import email.headerregistry
import re import re
import secrets
from typing import Optional, Tuple from typing import Optional, Tuple
import mo import mo
import mo.config as config
import mo.db as db import mo.db as db
import mo.util import mo.util
from mo.util import logger from mo.util import logger
...@@ -49,11 +52,13 @@ def validate_and_find_school(kod: str) -> db.Place: ...@@ -49,11 +52,13 @@ def validate_and_find_school(kod: str) -> db.Place:
return place return place
def find_or_create_user(email: str, krestni: str, prijmeni: str, is_org: bool, reason: str) -> Tuple[db.User, bool]: def find_or_create_user(email: str, krestni: Optional[str], prijmeni: Optional[str], is_org: bool, reason: str) -> Tuple[db.User, bool]:
sess = db.get_session() sess = db.get_session()
user = sess.query(db.User).filter_by(email=email).one_or_none() user = sess.query(db.User).filter_by(email=email).one_or_none()
is_new = user is None is_new = user is None
if user is None: # HACK: Podmínku je nutné zapsat znovu místo užití is_new, jinak si s tím mypy neporadí if user is None: # HACK: Podmínku je nutné zapsat znovu místo užití is_new, jinak si s tím mypy neporadí
if not krestni or not prijmeni:
raise mo.CheckError('Osoba s daným emailem zatím neexistuje, je nutné uvést její jméno.')
user = db.User(email=email, first_name=krestni, last_name=prijmeni, is_org=is_org) user = db.User(email=email, first_name=krestni, last_name=prijmeni, is_org=is_org)
sess.add(user) sess.add(user)
sess.flush() # Aby uživatel dostal user_id sess.flush() # Aby uživatel dostal user_id
...@@ -64,7 +69,7 @@ def find_or_create_user(email: str, krestni: str, prijmeni: str, is_org: bool, r ...@@ -64,7 +69,7 @@ def find_or_create_user(email: str, krestni: str, prijmeni: str, is_org: bool, r
details={'action': 'create-user', 'reason': reason, 'new': db.row2dict(user)}, details={'action': 'create-user', 'reason': reason, 'new': db.row2dict(user)},
) )
else: else:
if user.first_name != krestni or user.last_name != prijmeni: if (krestni and user.first_name != krestni) or (prijmeni and user.last_name != prijmeni):
raise mo.CheckError(f'Osoba již registrována s odlišným jménem {user.full_name()}') raise mo.CheckError(f'Osoba již registrována s odlišným jménem {user.full_name()}')
if (user.is_admin or user.is_org) != is_org: if (user.is_admin or user.is_org) != is_org:
if is_org: if is_org:
...@@ -74,11 +79,17 @@ def find_or_create_user(email: str, krestni: str, prijmeni: str, is_org: bool, r ...@@ -74,11 +79,17 @@ def find_or_create_user(email: str, krestni: str, prijmeni: str, is_org: bool, r
return user, is_new return user, is_new
def find_or_create_participant(user: db.User, year: int, school_id: int, birth_year: int, grade: str, reason: str) -> Tuple[db.Participant, bool]: def find_or_create_participant(user: db.User, year: int, school_id: Optional[int], birth_year: Optional[int], grade: Optional[str], reason: str) -> Tuple[db.Participant, bool]:
sess = db.get_session() sess = db.get_session()
part = sess.query(db.Participant).get((user.user_id, year)) part = sess.query(db.Participant).get((user.user_id, year))
is_new = part is None is_new = part is None
if part is None: if part is None:
if not school_id:
raise mo.CheckError('Osoba s daným emailem zatím není zaregistrovaná do ročníku, je nutné uvést školu.')
if not birth_year:
raise mo.CheckError('Osoba s daným emailem zatím není zaregistrovaná do ročníku, je nutné uvést rok narození.')
if not grade:
raise mo.CheckError('Osoba s daným emailem zatím není zaregistrovaná do ročníku, je nutné uvést ročník.')
part = db.Participant(user=user, year=year, school=school_id, birth_year=birth_year, grade=grade) part = db.Participant(user=user, year=year, school=school_id, birth_year=birth_year, grade=grade)
sess.add(part) sess.add(part)
logger.info(f'{reason.title()}: Založen účastník #{user.user_id}') logger.info(f'{reason.title()}: Založen účastník #{user.user_id}')
...@@ -88,9 +99,9 @@ def find_or_create_participant(user: db.User, year: int, school_id: int, birth_y ...@@ -88,9 +99,9 @@ def find_or_create_participant(user: db.User, year: int, school_id: int, birth_y
details={'action': 'create-participant', 'reason': reason, 'new': db.row2dict(part)}, details={'action': 'create-participant', 'reason': reason, 'new': db.row2dict(part)},
) )
else: else:
if (part.school != school_id if ((school_id and part.school != school_id)
or part.grade != grade or (grade and part.grade != grade)
or part.birth_year != birth_year): or (birth_year and part.birth_year != birth_year)):
raise mo.CheckError('Účastník již zaregistrován s odlišnou školou/ročníkem/rokem narození') raise mo.CheckError('Účastník již zaregistrován s odlišnou školou/ročníkem/rokem narození')
return part, is_new return part, is_new
...@@ -107,7 +118,7 @@ def find_or_create_participation(user: db.User, contest: db.Contest, place: Opti ...@@ -107,7 +118,7 @@ def find_or_create_participation(user: db.User, contest: db.Contest, place: Opti
is_new = pions == [] is_new = pions == []
if is_new: if is_new:
pion = db.Participation(user=user, contest=contest, place_id=place.place_id, state=db.PartState.invited) pion = db.Participation(user=user, contest=contest, place_id=place.place_id, state=db.PartState.active)
sess.add(pion) sess.add(pion)
logger.info(f'{reason.title()}: Založena účast user=#{user.user_id} contest=#{contest.contest_id} place=#{place.place_id}') logger.info(f'{reason.title()}: Založena účast user=#{user.user_id} contest=#{contest.contest_id} place=#{place.place_id}')
mo.util.log( mo.util.log(
...@@ -165,17 +176,23 @@ def user_by_uid(uid: int) -> db.User: ...@@ -165,17 +176,23 @@ def user_by_uid(uid: int) -> db.User:
return db.get_session().query(db.User).get(uid) return db.get_session().query(db.User).get(uid)
def set_password(user: db.User, passwd: str): password_help = 'Heslo musí mít alespoň 8 znaků. Doporučujeme kombinovat velká a malá písmena a číslice.'
def validate_password(passwd: str) -> bool:
return len(passwd) >= 8
def set_password(user: db.User, passwd: str, reset: bool = False):
salt = bcrypt.gensalt() salt = bcrypt.gensalt()
hashed = bcrypt.hashpw(passwd.encode('utf-8'), salt) hashed = bcrypt.hashpw(passwd.encode('utf-8'), salt)
user.password_hash = hashed.decode('us-ascii') user.password_hash = hashed.decode('us-ascii')
user.reset_at = None if reset:
user.last_login_at = datetime.datetime.now() user.reset_at = mo.now
mo.util.log( mo.util.log(
type=db.LogType.user, type=db.LogType.user,
what=user.user_id, what=user.user_id,
details={"action": "set-passwd"}, details={'action': 'do-reset'},
) )
...@@ -185,63 +202,68 @@ def check_password(user: db.User, passwd: str): ...@@ -185,63 +202,68 @@ def check_password(user: db.User, passwd: str):
def login(user: db.User): def login(user: db.User):
user.last_login_at = datetime.datetime.now() user.last_login_at = mo.now
user.reset_at = None
def ask_reset_password(user: db.User) -> str:
user.reset_at = datetime.datetime.now()
when = int(user.reset_at.timestamp())
token = mo.tokens.sign_token([str(user.user_id), str(when)], 'reset')
mo.util.log(
type=db.LogType.user,
what=user.user_id,
details={'action': 'ask-reset'},
)
return token
def make_activation_token(user: db.User) -> str:
user.reset_at = mo.now
when = int(mo.now.timestamp())
return mo.tokens.sign_token([str(user.user_id), str(when)], 'activate')
def check_reset_password(token: str) -> Optional[db.User]:
# Někteří klienti při kopírování adresy z mailu do prohlížeče
# přidávají divné Unicodové znaky (přepnutí směru psaní atd., viz issue #58).
token = re.sub(r'[^!-~]', "", token)
fields = mo.tokens.verify_token(token, 'reset') def check_activation_token(token: str) -> Optional[db.User]:
token = mo.util.clean_up_token(token)
fields = mo.tokens.verify_token(token, 'activate')
if not fields or len(fields) != 2: if not fields or len(fields) != 2:
return None return None
user = db.get_session().query(db.User).filter_by(user_id=int(fields[0])).first() user_id = int(fields[0])
token_time = datetime.datetime.fromtimestamp(int(fields[1]), tz=dateutil.tz.UTC)
if user.password_hash is None: user = user_by_uid(user_id)
reset_token_validity_time = datetime.timedelta(days=28) if not user:
return None
elif token_time < mo.now - datetime.timedelta(days=28):
return None
else: else:
reset_token_validity_time = datetime.timedelta(hours=24) return user
now = datetime.datetime.now().astimezone()
if (user def new_reg_request(type: db.RegReqType, client: str) -> Optional[db.RegRequest]:
and user.reset_at is not None sess = db.get_session()
and fields[1] == str(int(user.reset_at.timestamp()))
and now - user.reset_at < reset_token_validity_time): # Zatím jen jednoduchý rate limit, časem možno vylepšit
return user in_last_minute = db.get_count(sess.query(db.RegRequest).filter(db.RegRequest.created_at >= mo.now - datetime.timedelta(minutes=1)))
else: if in_last_minute >= config.REG_MAX_PER_MINUTE:
return None return None
email_token = mo.tokens.sign_token([str(int(mo.now.timestamp())), secrets.token_hex(16)], 'reg-request')
def cancel_reset_password(user: db.User): return db.RegRequest(
user.reset_at = None type=type,
mo.util.log( created_at=mo.now,
type=db.LogType.user, expires_at=mo.now + datetime.timedelta(minutes=config.REG_TOKEN_VALIDITY),
what=user.user_id, email_token=email_token,
details={'action': 'cancel-reset'}, client=client,
) )
def do_reset_password(user: db.User): def expire_reg_requests():
user.reset_at = None sess = db.get_session()
conn = sess.connection()
table = db.RegRequest.__table__
conn.execute(table.delete().where(table.c.expires_at < mo.now))
sess.commit()
def request_reset_password(user: db.User, client: str) -> Optional[db.RegRequest]:
logger.info('Login: Požadavek na reset hesla pro <%s>', user.email)
rr = new_reg_request(db.RegReqType.reset_passwd, client)
if rr:
db.get_session().add(rr)
rr.user_id = user.user_id
mo.util.log( mo.util.log(
type=db.LogType.user, type=db.LogType.user,
what=user.user_id, what=user.user_id,
details={'action': 'do-reset'}, details={'action': 'ask-reset'},
) )
return rr
...@@ -4,18 +4,13 @@ from dataclasses import dataclass ...@@ -4,18 +4,13 @@ from dataclasses import dataclass
import datetime import datetime
import decimal import decimal
import dateutil.tz import dateutil.tz
import email.message
import email.headerregistry
import locale import locale
import logging import logging
import os import os
import re import re
import secrets import secrets
import subprocess
import sys import sys
from typing import Any, Optional, NoReturn, Tuple, List from typing import Any, Optional, NoReturn, Tuple, List
import textwrap
import urllib.parse
import mo import mo
import mo.db as db import mo.db as db
...@@ -60,88 +55,6 @@ def log(type: db.LogType, what: int, details: Any): ...@@ -60,88 +55,6 @@ def log(type: db.LogType, what: int, details: Any):
db.get_session().add(entry) db.get_session().add(entry)
def send_user_email(user: db.User, subject: str, body: str) -> bool:
logger.info(f'Mail: "{subject}" -> {user.email}')
mail_from = getattr(config, 'MAIL_FROM', None)
if mail_from is None:
logger.error('Mail: V configu chybí nastavení MAIL_FROM')
return False
msg = email.message.EmailMessage()
msg['From'] = email.headerregistry.Address(
display_name='Odevzdávací Systém MO',
addr_spec=mail_from,
)
msg['To'] = [
email.headerregistry.Address(
display_name=user.full_name(),
addr_spec=user.email,
)
]
msg['Reply-To'] = email.headerregistry.Address(
display_name='Správce OSMO',
addr_spec=config.MAIL_CONTACT,
)
msg['Subject'] = 'OSMO – ' + subject
msg['Date'] = datetime.datetime.now()
msg.set_content(body, cte='quoted-printable')
mail_instead = getattr(config, 'MAIL_INSTEAD', None)
if mail_instead is None:
send_to = user.email
else:
send_to = mail_instead
sm = subprocess.Popen(
[
'/usr/sbin/sendmail',
'-oi',
'-f',
mail_from,
send_to,
],
stdin=subprocess.PIPE,
)
sm.communicate(msg.as_bytes())
if sm.returncode != 0:
logger.error('Mail: Sendmail failed with return code {}'.format(sm.returncode))
return False
return True
def password_reset_url(token: str) -> str:
return config.WEB_ROOT + 'auth/reset?' + urllib.parse.urlencode({'token': token}, safe=':')
def send_new_account_email(user: db.User, token: str) -> bool:
return send_user_email(user, 'Založen nový účet', textwrap.dedent('''\
Vítejte!
Právě Vám byl založen účet v Odevzdávacím systému Matematické olympiády.
Nastavte si prosím heslo na následující stránce:
{}
Váš OSMO
'''.format(password_reset_url(token))))
def send_password_reset_email(user: db.User, token: str) -> bool:
return send_user_email(user, 'Obnova hesla', textwrap.dedent('''\
Někdo (pravděpodobně Vy) požádal o obnovení hesla k Vašemu účtu v Odevzdávacím
systému Matematické olympiády. Heslo si můžete nastavit, případně požadavek
zrušit, na následující stránce:
{}
Váš OSMO
'''.format(password_reset_url(token))))
def die(msg: str) -> NoReturn: def die(msg: str) -> NoReturn:
print(msg, file=sys.stderr) print(msg, file=sys.stderr)
sys.exit(1) sys.exit(1)
...@@ -268,3 +181,9 @@ def parse_int_list(a: str, maxim: int = 200) -> List[int]: ...@@ -268,3 +181,9 @@ def parse_int_list(a: str, maxim: int = 200) -> List[int]:
raise mo.CheckError("Větší číslo nemůže být před menším") raise mo.CheckError("Větší číslo nemůže být před menším")
r += [c[0]] if len(c) == 1 else range(c[0], c[1] + 1) r += [c[0]] if len(c) == 1 else range(c[0], c[1] + 1)
return r return r
def clean_up_token(token: str) -> str:
# Někteří klienti při kopírování adresy z mailu do prohlížeče
# přidávají divné Unicodové znaky (přepnutí směru psaní atd., viz issue #58).
return re.sub(r'[^!-~]', "", token)
...@@ -41,6 +41,13 @@ def timeformat(dt: datetime) -> str: ...@@ -41,6 +41,13 @@ def timeformat(dt: datetime) -> str:
return dt.astimezone().strftime("%Y-%m-%d %H:%M") return dt.astimezone().strftime("%Y-%m-%d %H:%M")
def timeformat_short(dt: datetime) -> str:
if dt is None:
return ''
else:
return dt.astimezone().strftime("%Y-%m-%d")
def timedelta(d: datetime, ref: Optional[datetime] = None, descriptive: bool = False) -> str: def timedelta(d: datetime, ref: Optional[datetime] = None, descriptive: bool = False) -> str:
"""Vyrábí česky formátované řetězece 'za 3 minuty', 'před 27 dny' a podobně """Vyrábí česky formátované řetězece 'za 3 minuty', 'před 27 dny' a podobně
z rozdílu daného datetime a referenčního času (například now). z rozdílu daného datetime a referenčního času (například now).
......
...@@ -115,6 +115,7 @@ app.assets.add_assets([ ...@@ -115,6 +115,7 @@ app.assets.add_assets([
'bootstrap.min.css', 'bootstrap.min.css',
'mo.css', 'mo.css',
'js/news-reloader.js', 'js/news-reloader.js',
'js/osmo.js',
]) ])
...@@ -141,7 +142,7 @@ def init_request(): ...@@ -141,7 +142,7 @@ def init_request():
if not user: if not user:
# Uživatel mezitím přestal existovat # Uživatel mezitím přestal existovat
app.logger.error('Zrušena session pro neexistujícího uživatele uid=%s', session['uid']) app.logger.error('Zrušena session pro neexistujícího uživatele uid=%s', session['uid'])
return mo.web.auth.logout() return mo.web.acct.logout()
else: else:
user = None user = None
...@@ -171,18 +172,28 @@ app.before_request(init_request) ...@@ -171,18 +172,28 @@ app.before_request(init_request)
### UWSGI glue ### ### UWSGI glue ###
# Čas od času se probudíme a spustíme garbage collector:
# - projdeme joby pro případ, že by se ztratil signál
# - expirujeme zastaralé joby
# - expirujeme zastaralé registrační tokeny
@app.cli.command('gc')
def gc():
"""Run garbage collector."""
mo.now = mo.util.get_now()
mo.jobs.process_jobs()
mo.users.expire_reg_requests()
try: try:
import uwsgi import uwsgi
from uwsgidecorators import timer, signal from uwsgidecorators import timer, signal
# Čas od času se probudíme a projdeme joby pro případ, že by se ztratil signál. @timer(config.GC_PERIOD, target='mule')
# Také při tom expirujeme zastaralé joby.
@timer(config.JOB_GC_PERIOD, target='mule')
def mule_timer(signum): def mule_timer(signum):
# app.logger.debug('Mule: Timer tick') # app.logger.debug('Mule: Timer tick')
with app.app_context(): with app.app_context():
mo.now = mo.util.get_now() garbage_collect()
mo.jobs.process_jobs()
# Obykle při vložení jobu dostaneme signál. # Obykle při vložení jobu dostaneme signál.
@signal(42, target='mule') @signal(42, target='mule')
...@@ -205,7 +216,8 @@ except ImportError: ...@@ -205,7 +216,8 @@ except ImportError:
# Většina webu je v samostatných modulech # Většina webu je v samostatných modulech
import mo.web.auth import mo.web.api
import mo.web.acct
import mo.web.jinja import mo.web.jinja
import mo.web.menu import mo.web.menu
import mo.web.misc import mo.web.misc
......
This diff is collapsed.
from flask import request
from flask.json import jsonify
from sqlalchemy import func
from sqlalchemy.orm import joinedload
import werkzeug.exceptions
import mo.db as db
from mo.util_format import inflect_with_number
from mo.web import app
@app.route('/api/')
def api_root():
"""Slouží jako prefix pro konstrukci URL v JavaScriptu."""
raise werkzeug.exceptions.NotFound()
@app.route('/api/find-town')
def api_find_town():
query = request.args.get('q')
if query is None or len(query) < 2:
return jsonify(error='Zadejte alespoň 2 znaky jména obce.')
elif '%' in query:
return jsonify(error='Nepovolené znaky ve jménu obce.')
else:
max_places = 50
places = (db.get_session().query(db.Place)
.filter_by(level=3)
.filter(func.lower(db.f_unaccent(db.Place.name)).like(func.lower(db.f_unaccent(query + '%'))))
.options(joinedload(db.Place.parent_place))
.order_by(db.Place.name, db.Place.place_id)
.limit(max_places)
.all())
if not places:
return jsonify(error='Nenalezena žádná obec.')
# XXX: Nemůže se stát, že nastane přesná shoda a k tomu příliš mnoho nepřesných?
if len(places) >= max_places:
return jsonify(error='Nalezeno příliš mnoho obcí. Zadejte prosím více znaků jména.')
res = []
for p in places:
name = p.name
if p.name != p.parent_place.name:
name += f' (okres {p.parent_place.name})'
res.append([p.place_id, name])
msg = inflect_with_number(len(res), 'Nalezena %s obec.', 'Nalezeny %s obce.', 'Nalezeno %s obcí.')
return jsonify(found=res, msg=msg)
@app.route('/api/get-schools')
def api_get_schools():
town = request.args.get('town')
if town is None or not town.isnumeric():
raise werkzeug.exceptions.BadRequest()
town_id = int(town)
places = (db.get_session().query(db.Place)
.filter_by(level=4, type=db.PlaceType.school, parent=town_id)
.options(joinedload(db.Place.school))
.order_by(db.Place.name)
.all())
zs = []
ss = []
for p in places:
s = {
'id': p.place_id,
'name': p.name,
}
if p.school.is_zs:
zs.append(s)
if p.school.is_ss:
ss.append(s)
return jsonify(zs=zs, ss=ss)
import datetime
from flask import render_template, request, g, redirect, url_for, session
from flask.helpers import flash
from flask_wtf import FlaskForm
import werkzeug.exceptions
import wtforms
from wtforms.fields.html5 import EmailField
import wtforms.validators as validators
from sqlalchemy.orm import joinedload
from typing import Optional
import mo.util
import mo.db as db
import mo.rights
import mo.users
from mo.web import app, NeedLoginError
class LoginForm(FlaskForm):
next = wtforms.HiddenField()
email = EmailField('E-mail', validators=[validators.DataRequired()])
passwd = wtforms.PasswordField('Heslo')
submit = wtforms.SubmitField('Přihlásit se')
reset = wtforms.SubmitField('Zapomenuté heslo')
def login_and_redirect(user: db.User, url: Optional[str] = None):
session.clear()
session['uid'] = user.user_id
if not url:
if user.is_admin or user.is_org:
url = url_for('org_index')
else:
url = url_for('index')
else:
url = request.script_root + url
return redirect(url)
@app.route('/auth/login', methods=('GET', 'POST'))
def login():
form = LoginForm(email=request.args.get('email'))
if not form.validate_on_submit():
return render_template('login.html', form=form, error=None)
email = form.email.data
user = mo.users.user_by_email(email)
if not user:
app.logger.error('Login: Neznámý uživatel <%s>', email)
flash('Neznámý uživatel', 'danger')
elif form.reset.data:
app.logger.info('Login: Požadavek na reset hesla pro <%s>', email)
min_time_between_resets = datetime.timedelta(minutes=1)
now = datetime.datetime.now().astimezone()
if (user.reset_at is not None
and now - user.reset_at < min_time_between_resets):
flash('Poslední požadavek na obnovení hesla byl odeslán příliš nedávno', 'danger')
else:
token = mo.users.ask_reset_password(user)
db.get_session().commit()
mo.util.send_password_reset_email(user, token)
flash('Na uvedenou adresu byl odeslán e-mail s odkazem na obnovu hesla', 'success')
elif not form.passwd.data or not mo.users.check_password(user, form.passwd.data):
app.logger.error('Login: Špatné heslo pro uživatele <%s>', email)
flash('Chybné heslo', 'danger')
else:
if user.is_admin:
typ = ' (admin)'
elif user.is_org:
typ = ' (org)'
elif user.is_test:
typ = ' (test)'
else:
typ = ""
app.logger.info('Login: Přihlásil se uživatel #%s <%s>%s', user.user_id, email, typ)
mo.users.login(user)
db.get_session().commit()
return login_and_redirect(user, url=form.next.data)
return render_template('login.html', form=form)
@app.route('/auth/logout', methods=('POST',))
def logout():
session.clear()
return redirect(url_for('index'))
@app.route('/auth/incarnate/<int:id>', methods=('POST',))
def incarnate(id):
if not g.user.is_admin:
raise werkzeug.exceptions.Forbidden()
new_user = db.get_session().query(db.User).get(id)
if not new_user:
raise werkzeug.exceptions.NotFound()
app.logger.info('Login: Uživatel #%s se převtělil na #%s', g.user.user_id, new_user.user_id)
return login_and_redirect(new_user)
@app.route('/user/settings')
def user_settings():
sess = db.get_session()
roles = []
if g.user:
roles = (sess.query(db.UserRole)
.filter_by(user_id=g.user.user_id)
.options(joinedload(db.UserRole.place))
.all())
return render_template('settings.html', roles=roles, roles_by_type=mo.rights.roles_by_type)
@app.errorhandler(NeedLoginError)
def handle_need_login(e):
form = LoginForm()
form.next.data = request.path
return render_template('login.html', form=form), e.code
class ResetForm(FlaskForm):
email = EmailField('E-mail', description='Účet pro který se nastavuje nové heslo', render_kw={"disabled": "disabled"})
token = wtforms.HiddenField()
passwd = wtforms.PasswordField('Nové heslo', description='Heslo musí mít alespoň 8 znaků. Doporučujeme kombinovat velká a malá písmena a číslice.')
submit = wtforms.SubmitField('Nastavit heslo')
cancel = wtforms.SubmitField('Zrušit obnovu hesla')
@app.route('/auth/reset', methods=('GET', 'POST'))
def reset():
token = request.args.get('token')
if not token:
flash('Žádný token pro resetování hesla', 'danger')
return redirect(url_for('login'))
user = mo.users.check_reset_password(token)
if not user:
flash('Neplatný požadavek na obnovu hesla', 'danger')
return redirect(url_for('login'))
form = ResetForm(token=token, email=user.email)
ok = form.validate_on_submit()
if not ok:
return render_template('reset.html', form=form)
if form.cancel.data:
mo.users.cancel_reset_password(user)
app.logger.info('Login: Zrušen reset hesla pro uživatele <%s>', user.email)
db.get_session().commit()
flash('Obnova hesla zrušena', 'warning')
return redirect(url_for('login'))
elif len(form.passwd.data) < 8:
flash('Heslo musí být aspoň 8 znaků dlouhé', 'danger')
return render_template('reset.html', form=form)
else:
mo.users.do_reset_password(user)
mo.users.set_password(user, form.passwd.data)
app.logger.info('Login: Reset hesla pro uživatele <%s>', user.email)
mo.util.log(
type=db.LogType.user,
what=user.user_id,
details={'action': 'reset-passwd'},
)
mo.users.login(user)
app.logger.info('Login: Přihlásil se uživatel <%s> po resetování hesla', user.email)
db.get_session().commit()
flash('Nastavení nového hesla a přihlášení do systému proběhlo úspěšně', 'success')
return login_and_redirect(user)
@app.errorhandler(werkzeug.exceptions.Forbidden)
def handle_forbidden(e):
return render_template('forbidden.html')