SpecialistOff.NET / Вопросы / Статьи / Фрагменты кода / Резюме / Метки / Помощь / Файлы

Скрипт добавления пользователя

Метки: [python];

• myapp - приложение
• app - Flask приложение
• lib - библиотека низкоуровневых функций
• models - модель SQLAlchemy

--- useradd.py

• useradd.py

#!/usr/bin/env python3

import sys
import argparse
from myapp import app, lib, models

def main():
    parser = argparse.ArgumentParser(description='Скрипт добавления пользователя',
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser._optionals.title = "Необязательные аргументы"

    parser.add_argument("--user", dest="user", help="Новый пользователь")
    parser.add_argument("--password", dest="password", help="Новый пароль")

    args = parser.parse_args()

    user = models.db_session.query(
        models.User
    ).filter(
        models.User.name==args.user
    ).first()
    if user:
        app.logger.warning('Пользователь %s уже существует' % args.user)
        sys.exit(1)
    user = models.User(args.user)
    user.password = lib.get_hash_password(args.password,
        app.config['SECRET_KEY'])
    models.db_session.add(user)
    models.db_session.commit()
    app.logger.info('Пользователь %s успешно добавлен' % args.user)

if __name__ == "__main__":
    try:
        main()
    except Exception as err:
        traceback.print_exc(file=sys.stdout)
        exit(1)

    exit(0)