def create(request):
"""
Create a django user object and a Lookout
account object.
"""
form = CreateAccountForm()
if request.method == "POST":
form = CreateAccountForm(request.POST)
if form.is_valid():
data = form.cleaned_data
user = User.objects.create_user(data['username'], '', data['password'])
user.save()
account = Account(email=data['email'],
user=user)
account.save()
user = authenticate(username=user.username, password=data['password'])
login(request, user)
return HttpResponseRedirect("/")
return render_to_response('account/create.html', {
'form' : form,
}, context_instance = RequestContext(request))
def setup():
user = User.objects.all()[0]
account = Account(user=user)
account.save()
client = Client(name="Beth", user=account)
client.save()
alert = Alert(user=account, client=client, term="l", type="l", frequency="50", length=50)
alert.save()
def create_account(email):
"""
Returns an account if it already
exists, otherwise creates it.
"""
acc = find_account(email)
if not acc:
acc = Account(email=email)
acc.save()
return acc
def setup_func():
user = User.objects.all()[0]
account = Account(user=user)
account.save()
client = Client(name="Beth",
user=account)
client.save()
LamsonState.objects.all().delete()
q = queue.Queue(email('run/error'))
q.clear()
q = queue.Queue(email('run/alerts'))
q.clear()
def save_acc_info(user_id, username, **kwargs):
"""Добавляем подключенный аккаунт Steam в БД.
Если аккаунт уже существует - обновляем данные, которые изменились.
"""
current_app.logger.info("Save account info function")
# Создаем список аргументов переданных функции
args = {}
for key, value in kwargs.items():
args[key] = value
db_steam_acc = Account.query.filter_by(username=username,
user_id=user_id).first()
if db_steam_acc:
current_app.logger.info(f"Account {username} already exists. "
f"Trying to update data.")
for arg, value in args.items():
# Если содержимое ячейки изменилось - перезаписываем его
if value is not None:
if getattr(db_steam_acc, arg) != value:
current_app.logger.info(
f"Old account {arg} is: "
f"{getattr(db_steam_acc, arg)}")
setattr(db_steam_acc, arg, value)
current_app.logger.info(
f"New account {arg} is: "
f"{getattr(db_steam_acc, arg)}")
else:
current_app.logger.info("Create account %s", username)
db_steam_acc = Account(
username=username,
user_id=user_id,
)
for arg, value in args.items():
setattr(db_steam_acc, arg, value)
try:
db.session.add(db_steam_acc)
except (DBAPIError, SQLAlchemyError) as err:
current_app.logger.info(err)
db.session.commit()
current_app.logger.info("Successful DB injection")
return None
def make_account(self, data, **kwargs):
"""Функция для десериализации объекта."""
return Account(**data)
