Compare commits
2 Commits
1ce84db59c
...
819aa6784f
Author | SHA1 | Date |
---|---|---|
Анатолий Богомолов | 819aa6784f | |
Анатолий Богомолов | a605795578 |
|
@ -1,4 +1,10 @@
|
|||
import csv
|
||||
from datetime import datetime
|
||||
import io
|
||||
import os
|
||||
|
||||
from flask import Blueprint, render_template, request
|
||||
from loguru import logger
|
||||
|
||||
from app.models.collection import Collection
|
||||
from app.extensions import db
|
||||
|
@ -39,3 +45,22 @@ def delete_collection(id: int):
|
|||
return "<small class=\"form-text\">Здесь ничего нет</small>", 200
|
||||
|
||||
return "", 200
|
||||
|
||||
@collections.route("/<int:id>/csv", methods=["GET"])
|
||||
def export_users_to_csv(id: int):
|
||||
collection: Collection = Collection.query.get_or_404(id)
|
||||
|
||||
if len(collection.users) <= 0:
|
||||
return "", 204
|
||||
|
||||
filename = f"{collection.name} - {datetime.now()}.csv"
|
||||
|
||||
fieldnames = collection.users[0].to_dict().keys()
|
||||
si = io.StringIO()
|
||||
|
||||
writer = csv.DictWriter(si, fieldnames)
|
||||
|
||||
writer.writeheader()
|
||||
writer.writerows(map(lambda user: user.to_dict(), collection.users))
|
||||
|
||||
return si.getvalue(), 200, {'Content-Disposition': f'attachment; filename={filename}', 'Content-type': 'text/csv'}
|
|
@ -4,7 +4,9 @@
|
|||
<ul class="list-group list-group-flush">
|
||||
<li class="list-group-item">Кол-во пользрвателей: {{ collection.users|length }}</li>
|
||||
</ul>
|
||||
<form action="/api/collections/{{ collection.id }}" method="DELETE" class="card-footer px-2 pt-2">
|
||||
<div class="card-footer px-2 pt-2">
|
||||
<div class="d-flex gap-2" role="group">
|
||||
<form action="/api/collections/{{ collection.id }}" method="DELETE">
|
||||
<button
|
||||
type="submit"
|
||||
hx-delete="/api/collections/{{ collection.id }}"
|
||||
|
@ -16,5 +18,13 @@
|
|||
Удалить
|
||||
</button>
|
||||
</form>
|
||||
<a
|
||||
href="/api/collections/{{ collection.id }}/csv"
|
||||
class="btn btn-outline-success"
|
||||
>
|
||||
Экспорт
|
||||
</a>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
|
@ -59,7 +59,7 @@ def delete_task(task_id: int):
|
|||
db.session.delete(task)
|
||||
db.session.commit()
|
||||
|
||||
return '', 204
|
||||
return '', 200
|
||||
|
||||
@tasks.route("/<int:task_id>/stop", methods=["PUT"])
|
||||
def stop_task(task_id: int):
|
||||
|
|
|
@ -20,3 +20,12 @@ class User(db.Model):
|
|||
|
||||
def exist(username: str, collection):
|
||||
return not User.query.filter_by(username=username, collection=collection).first() is None
|
||||
|
||||
def to_dict(self):
|
||||
return {
|
||||
'id': self.id,
|
||||
'first_name': self.first_name,
|
||||
'last_name': self.last_name,
|
||||
'phone': self.phone,
|
||||
'username': self.username,
|
||||
}
|
Loading…
Reference in New Issue