def post(self):
hacknight_schema = HacknightSchema()
json_data = request.get_json(force=True)
if not json_data:
return {
"message": "No input data provided"
}, HTTPStatus.BAD_REQUEST
try:
data = hacknight_schema.load(json_data)
except ValidationError as err:
return (err.messages), HTTPStatus.BAD_REQUEST
hacknight = Hacknight.query.filter_by(date=data["date"]).first()
if hacknight:
return {
"message": "Hacknight already exists."
}, HTTPStatus.CONFLICT
hacknight = Hacknight(date=data["date"])
db.session.add(hacknight)
db.session.commit()
return (
{
"message": "Hacknight created successfully.",
"hacknight": data
},
HTTPStatus.CREATED,
)
def test_add_new_hacknight_to_db(_db, new_hacknight):
"""Test addind hacknight do DB with valid data."""
db = _db
assert len(db.session.query(Hacknight).all()) == 0
new_hacknight = Hacknight(date=new_hacknight["date"])
db.session.add(new_hacknight)
db.session.commit()
hacknight = db.session.query(Hacknight).all()[0]
assert hacknight.date == datetime.date(2000, 10, 10)
def test_create_hacknight_with_same_date(auth_client, new_hacknight, _db):
"""Test add hcknight with the same date."""
db = _db
new_hacknight = Hacknight(date=new_hacknight["date"])
db.session.add(new_hacknight)
db.session.commit()
payload = {"date": "2000-10-10"}
rv = auth_client.post(
f"/hacknights/",
data=json.dumps(payload),
)
response = rv.get_json()
assert rv.status_code == HTTPStatus.CONFLICT
assert response["message"] == "Hacknight already exists."
def test_create_new_hacknight(new_hacknight):
"""Test hacknight model."""
new_hacknight = Hacknight(date=new_hacknight["date"])
assert new_hacknight.date == datetime.date(2000, 10, 10)
def import_attendance_list():
"""Import attendance list."""
rows = []
list_of_participants = []
# open file from exported csv
with open("./temp/list.csv") as csv_file:
csv_reader = csv.reader(csv_file, delimiter=",")
for row in csv_reader:
rows.append(row)
# get dates of historical hacknights
dates = rows[0][5:]
click.echo(f"Created list of dates: {dates}")
# skip header
rows = rows[1:]
# prepare users' data
for item in rows:
users_dict = {}
_id, github, name, phone, email, *_ = item
try:
first_name = name.split()[0]
last_name = name.split()[1]
except:
first_name = "Unknown"
last_name = "Unknown"
# skip when _id == ''
try:
users_dict["id"] = int(_id)
except:
continue
users_dict["github"] = github
users_dict["first_name"] = first_name
users_dict["last_name"] = last_name
users_dict["email"] = email
users_dict["phone"] = phone
list_of_participants.append(users_dict)
# create hacknights
for idx, date in enumerate(dates, 1):
day, month, year = date.split(".")
day, month = day.zfill(2), month.zfill(2)
hacknight = Hacknight(id=idx, date=f"{year}-{month}-{day}")
db.session.add(hacknight)
db.session.commit()
click.echo(f"Imported {len(dates)} hacknights")
# create participants
for idx, participant in enumerate(list_of_participants):
if participant["email"] == "":
participant["email"] = f"unknown_email_{idx}"
if participant["github"] == "":
participant["github"] = f"unknown_github_{idx}"
new_participant = Participant(**participant)
db.session.add(new_participant)
db.session.commit()
click.echo(f"Imported {len(list_of_participants)} participants")
# assign attendance list
for participant_idx, row in enumerate(rows, 1):
participant = Participant.query.get(participant_idx)
for hacknight_index, element in enumerate(row[5:], 1):
if "1" in element and participant:
hacknight = Hacknight.query.get(hacknight_index)
hacknight.participants.append(participant)
email = participant.email if participant else "???"
click.echo(f"Added {email} to hacknight {hacknight.date}")
db.session.flush()
db.session.commit()
