app.files

Complexity

Total Complexity

25

Size/Duplication

Total Lines	68
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
F	upload()	0	35	21
A	export()	0	17	4

from app import db
from datetime import datetime, date
import pandas as pd
import os
from natsort import index_natsorted
import numpy as np
from dateutil.relativedelta import relativedelta
from pathlib import Path
from io import TextIOWrapper
import csv
from .models import Schedule


def export():
    try:
        engine = db.create_engine(os.environ.get('DATABASE_URL')).connect()
    except:
        engine = db.create_engine('sqlite:///db.sqlite').connect()

    # pull the schedule information
    df = pd.read_sql('SELECT * FROM schedule;', engine)
    df = df.sort_values(by="startdate",
                        key=lambda x: np.argsort(index_natsorted(df["startdate"]))).reset_index(drop=True)

    csv_data = "Name,Amount,Type,Frequency,Next Date\n"
    for i in range(len(df.index)):
        # Create a CSV string from the data
        csv_data += f"{df['name'][i]},{df['amount'][i]},{df['type'][i]},{df['frequency'][i]},{df['startdate'][i]}\n"

    return csv_data


def upload(csv_file):
    csv_file = TextIOWrapper(csv_file, encoding='utf-8')
    csv_reader = csv.reader(csv_file, delimiter=',')
    next(csv_reader)

    format = '%Y-%m-%d %H:%M:%S'
    for row in csv_reader:
        try:
            name = row[0]
            amount = float(row[1])
            type = row[2]
            frequency = row[3]
            next_date = row[4]
            next_date = datetime.strptime(next_date, format).date()

            existing = Schedule.query.filter_by(name=name).first()

            if (not existing and (type == "Income" or type == "Expense")
                    and (frequency == "Monthly" or frequency == "Quarterly" or frequency == "Yearly" or
                         frequency == "Weekly" or frequency == "BiWeekly" or frequency == "Onetime")):
                schedule = Schedule(name=name, amount=amount, type=type, frequency=frequency, startdate=next_date)
                db.session.add(schedule)
                db.session.commit()
            elif (existing and (type == "Income" or type == "Expense")
                    and (frequency == "Monthly" or frequency == "Quarterly" or frequency == "Yearly" or
                         frequency == "Weekly" or frequency == "BiWeekly" or frequency == "Onetime")):
                existing.amount = amount
                existing.frequency = frequency
                existing.startdate = next_date
                existing.type = type
                db.session.commit()
        except:
            pass

    return 0