def import_expenses(self):
try:
row = int(self.le_initial_row.text()) - 1
description_col = int(self.le_description_col.text()) - 1
date_col = int(self.le_date_col.text()) - 1
default_date = self.default_date.date()
date_format = self.le_date_format.text()
percent = 1.0 / len(list(self.current_group.getMembers()))
while self.model.item(row) is not None:
print("Row ", row)
expense = Expense()
expense.setDescription(
self.model.item(row, description_col).text())
print("Expense: ", expense.getDescription())
date = QDate.fromString(
self.model.item(row, date_col).text(), date_format)
if not date.isValid():
date = default_date
expense.setDate(date.toString(Qt.ISODate))
print("Date: ", expense.getDate())
cost = 0
users = []
for member in self.current_group.getMembers():
print("Processing member ", member.getFirstName())
member_column = int(
self.member_widget_map[member.getId()].text()) - 1
paid = 0
try:
paid = float(
self.model.item(row, member_column).text())
print("Expense: ",
self.model.item(row, member_column).text())
except:
pass
cost = cost + paid
expense_user = ExpenseUser()
expense_user.setId(member.getId())
expense_user.setPaidShare(str(paid))
users.append(expense_user)
for expense_user in users:
expense_user.setOwedShare(str(cost * percent))
if cost == 0:
raise Exception(
self.tr('No se ha introducido monto para el gasto'))
expense.setCost(str(cost))
expense.setUsers(users)
expense.setGroupId(self.current_group.id)
self.sObj.createExpense(expense)
row = row + 1
self.le_initial_row.setText(str(row + 1))
except Exception as inst:
QMessageBox.critical(
self, self.tr("Error"),
self.tr("Se ha producido un error en la fila") + str(row + 1) +
"\n" + str(inst))
traceback.print_exc()
def test_expense_flow(self):
receipt = os.path.join(os.path.dirname(os.path.realpath(__file__)),
"receipt.jpg")
expense = Expense()
expense.setDescription("End2EndTest")
expense.setCost('10')
expense.setGroupId(19571167)
expense.setSplitEqually()
expense.setReceipt(receipt)
# create expense
expense, error = self.sObj.createExpense(expense)
self.assertIsNone(error)
self.assertIsNotNone(expense.getId())
# delete expense
success, error = self.sObj.deleteExpense(expense.getId())
self.assertIsNone(error)
self.assertTrue(success)
def _create_expense(initiator_id: str, recipient_id: str, amount: float, description: str):
"""
This will create an expense where the {initiator} owes the {recipient} {amount}.
"""
if amount <= 0:
raise CommandException('Negative amounts not allowed')
tso: User = User.objects.get(friendly_name='tso')
initiator: User = User.objects.get(slack_id=initiator_id)
recipient: User = User.objects.get(slack_id=recipient_id)
users = []
expense = Expense()
expense.setGroupId(settings.SPLITWISE_GROUP_ID)
expense.setDescription(description)
expense.setCost(amount)
initiator_user = ExpenseUser()
initiator_user.setId(initiator.splitwise_id)
initiator_user.setPaidShare(0)
initiator_user.setOwedShare(amount)
users.append(initiator_user)
recipient_user = ExpenseUser()
recipient_user.setId(recipient.splitwise_id)
recipient_user.setPaidShare(amount)
recipient_user.setOwedShare(0)
users.append(recipient_user)
# We do this because we're using my access token.
if tso.id not in [initiator.id, recipient.id]:
tso_user = ExpenseUser()
tso_user.setId(tso.splitwise_id)
tso_user.setPaidShare(0)
tso_user.setOwedShare(0)
users.append(tso_user)
expense.setUsers(users)
if not sObj.createExpense(expense):
raise CommandException('Failed to create SW expense')
def transaction(Description, Group, Payer, Price, Contributors):
# price, date, description, group ID or this particular groups name, members
sObj = Splitwise(consumer_key, secret_key)
sObj.setAccessToken(session['access_token'])
user = sObj.getCurrentUser()
groups = sObj.getGroups()
group_dict = {group.getName(): group for group in groups}
importedData = {'Date': datetime.datetime.now(), 'Description': Description, 'Group': Group, 'Payer': Payer, 'Debit': Price}
expense = Expense()
price = float(importedData['Debit'] or 0)
expense.setCost(price)
expense.setDate(importedData['Date'])
expense.setDescription(importedData['Description'])
expense.setGroupId(group_dict[importedData['Group']].getId())
members = group_dict[importedData['Group']].getMembers()
users = []
contributors = Contributors
for member in members:
if member.getFirstName() in contributors:
user = ExpenseUser()
user.setId(member.getId())
if member.getFirstName() == importedData['Payer']:
user.setPaidShare(price)
else:
user.setPaidShare(0)
users.append(user)
paid = 0
share = round(price/len(users), 2)
for user in users:
user.setOwedShare(share)
paid = paid + share
diff = price - paid
if diff != 0:
user = random.choice(users)
user.setOwedShare(share + diff)
expense.setUsers(users)
expense = sObj.createExpense(expense)
def submission():
global df
# TODO: add checks beforehand whether the user has clicked on all checkboxes
# print(request.form)
sObj = Splitwise(Config.consumer_key,Config.consumer_secret)
sObj.setAccessToken(session['access_token'])
saleh = ExpenseUser()
saleh.setId(2242086)
paypal = ExpenseUser()
paypal.setId(18572820)
nuraini = ExpenseUser()
nuraini.setId(2705458)
for key in request.form:
if not is_number(key):
continue
value = request.form[key]
if value == 'Payment':
continue # we dont handle payments yet
print(key, value)
amount = df.iloc[int(float(key))]['Amount']
users = []
users.append(saleh)
users.append(paypal)
users.append(nuraini)
saleh.setPaidShare('0.00')
nuraini.setPaidShare('0.00')
paypal.setPaidShare('0.00')
saleh.setOwedShare('0.00')
nuraini.setOwedShare('0.00')
paypal.setOwedShare('0.00')
expense = Expense()
expense.setUsers(users)
expense.setGroupId(6456733)
expense.setCost(str(abs(float(amount))))
expense.setDescription(df.iloc[int(float(key))]['Description'])
try:
expense.setDate(datetime.datetime.strptime(df.iloc[int(float(key))]['Trans Date'], '%m/%d/%Y').strftime('%d/%m/%Y'))
except:
expense.setDate(datetime.datetime.strptime(df.iloc[int(float(key))]['Trans Date'], '%m/%d/%y').strftime('%d/%m/%Y'))
# case where a transaction is refunded
if float(amount) > 0:
if value == 'Saleh':
paypal.setOwedShare(str(abs(float(amount))))
saleh.setPaidShare(str(abs(float(amount))))
elif value == 'Nuraini':
paypal.setOwedShare(str(abs(float(amount))))
nuraini.setPaidShare(str(abs(float(amount))))
expense = sObj.createExpense(expense)
continue
# case for expenses
if value == 'Saleh':
saleh.setOwedShare(str(abs(float(amount))))
paypal.setPaidShare(str(abs(float(amount))))
expense = sObj.createExpense(expense)
elif value == 'Nuraini':
nuraini.setOwedShare(str(abs(float(amount))))
paypal.setPaidShare(str(abs(float(amount))))
expense = sObj.createExpense(expense)
elif value == 'Half-Split':
half = round(abs(float(amount))/2,2)
other_half = abs(float(amount))-half
nuraini.setOwedShare(half)
saleh.setOwedShare(other_half)
paypal.setPaidShare(str(abs(float(amount))))
expense = sObj.createExpense(expense)
elif value == 'Share':
continue
return redirect(url_for("success"))
def test_updateExpense_split_equally_success(self, mockMakeRequest):
mockMakeRequest.return_value = '{"expenses":[{"id":1010906976,"group_id":19433671,"friendship_id":null,"expense_bundle_id":null,"description":"Testing","repeats":false,"repeat_interval":"never","email_reminder":false,"email_reminder_in_advance":-1,"next_repeat":null,"details":"Full details of the expense","comments_count":0,"payment":false,"creation_method":null,"transaction_method":"offline","transaction_confirmed":false,"transaction_id":null,"cost":"13.0","currency_code":"SGD","repayments":[{"from":643871,"to":79774,"amount":"5.0"}],"date":"2020-06-24T06:57:29Z","created_at":"2020-06-24T06:57:29Z","created_by":{"id":79774,"first_name":"Naman","last_name":"Aggarwal","picture":{"medium":"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"},"custom_picture":true},"updated_at":"2020-06-24T06:57:29Z","updated_by":null,"deleted_at":null,"deleted_by":null,"category":{"id":18,"name":"General"},"receipt":{"large":null,"original":null},"users":[{"user":{"id":79774,"first_name":"Naman","last_name":"Aggarwal","picture":{"medium":"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"}},"user_id":79774,"paid_share":"10.0","owed_share":"5.0","net_balance":"5.0"},{"user":{"id":643871,"first_name":"Shantanu","last_name":"Alshi","picture":{"medium":"https://s3.amazonaws.com/splitwise/uploads/user/default_avatars/avatar-blue19-100px.png"}},"user_id":643871,"paid_share":"0.0","owed_share":"5.0","net_balance":"-5.0"}]}],"errors":{}}' # noqa: E501
expense = Expense()
expense.id = 1010906976
expense.setCost('13')
expense.setDescription("Testing")
expense.setGroupId(19433671)
expense.setDetails("Full details of the expense")
expense.setSplitEqually()
expenseRes, errors = self.sObj.updateExpense(expense)
mockMakeRequest.assert_called_with(
"https://secure.splitwise.com/api/v3.0/update_expense/1010906976",
"POST", {
'cost': '13',
'description': 'Testing',
'details': 'Full details of the expense',
'group_id': 19433671,
'split_equally': True
},
files=None)
self.assertIsNone(errors)
self.assertEqual(expenseRes.getId(), 1010906976)
self.assertEqual(expenseRes.getGroupId(), 19433671)
self.assertEqual(expenseRes.getFriendshipId(), None)
self.assertEqual(expenseRes.getExpenseBundleId(), None)
self.assertEqual(expenseRes.getDescription(), "Testing")
self.assertEqual(expenseRes.getRepeatInterval(), "never")
self.assertEqual(expenseRes.getEmailReminderInAdvance(), -1)
self.assertEqual(expenseRes.getNextRepeat(), None)
self.assertEqual(expenseRes.getDetails(),
"Full details of the expense")
self.assertEqual(expenseRes.getCommentsCount(), 0)
self.assertEqual(expenseRes.getCreationMethod(), None)
self.assertEqual(expenseRes.getTransactionMethod(), "offline")
self.assertEqual(expenseRes.getTransactionId(), None)
self.assertEqual(expenseRes.getCost(), "13.0")
self.assertEqual(expenseRes.getCurrencyCode(), "SGD")
self.assertEqual(len(expenseRes.getRepayments()), 1)
self.assertEqual(expenseRes.getRepayments()[0].getFromUser(), 643871)
self.assertEqual(expenseRes.getRepayments()[0].getToUser(), 79774)
self.assertEqual(expenseRes.getRepayments()[0].getAmount(), "5.0")
self.assertEqual(expenseRes.getDate(), "2020-06-24T06:57:29Z")
self.assertEqual(expenseRes.getCreatedAt(), "2020-06-24T06:57:29Z")
self.assertEqual(expenseRes.getCreatedBy().getId(), 79774)
self.assertEqual(expenseRes.getCreatedBy().getFirstName(), "Naman")
self.assertEqual(expenseRes.getCreatedBy().getLastName(), "Aggarwal")
self.assertEqual(
expenseRes.getCreatedBy().getPicture().getMedium(),
"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"
)
self.assertEqual(expenseRes.getUpdatedAt(), "2020-06-24T06:57:29Z")
self.assertEqual(expenseRes.getUpdatedBy(), None)
self.assertEqual(expenseRes.getDeletedAt(), None)
self.assertEqual(expenseRes.getDeletedBy(), None)
self.assertEqual(expenseRes.getCategory().getId(), 18)
self.assertEqual(expenseRes.getCategory().getName(), "General")
self.assertEqual(expenseRes.getReceipt().getLarge(), None)
self.assertEqual(expenseRes.getReceipt().getOriginal(), None)
self.assertEqual(len(expenseRes.getUsers()), 2)
self.assertEqual(expenseRes.getUsers()[0].getId(), 79774)
self.assertEqual(expenseRes.getUsers()[0].getFirstName(), "Naman")
self.assertEqual(expenseRes.getUsers()[0].getLastName(), "Aggarwal")
self.assertEqual(
expenseRes.getUsers()[0].getPicture().getMedium(),
"https://splitwise.s3.amazonaws.com/uploads/user/avatar/79774/medium_mypic.jpg"
)
self.assertEqual(expenseRes.getUsers()[0].getPaidShare(), "10.0")
self.assertEqual(expenseRes.getUsers()[0].getOwedShare(), "5.0")
self.assertEqual(expenseRes.getUsers()[0].getNetBalance(), "5.0")
self.assertEqual(expenseRes.getUsers()[1].getId(), 643871)
self.assertEqual(expenseRes.getUsers()[1].getFirstName(), "Shantanu")
self.assertEqual(expenseRes.getUsers()[1].getLastName(), "Alshi")
self.assertEqual(
expenseRes.getUsers()[1].getPicture().getMedium(),
"https://s3.amazonaws.com/splitwise/uploads/user/default_avatars/avatar-blue19-100px.png"
)
self.assertEqual(expenseRes.getUsers()[1].getPaidShare(), "0.0")
self.assertEqual(expenseRes.getUsers()[1].getOwedShare(), "5.0")
self.assertEqual(expenseRes.getUsers()[1].getNetBalance(), "-5.0")
def addExpense(self, cost: float, description: str, group_id: int,
payer: str) -> None:
"""
Adds expense to Splitwise group. If expenses don't evenly get
distributed, it will randomly assign pennies to even things off
"""
expense = Expense()
expense.setCost(str(cost))
expense.setDescription(description)
expense.setGroupId(group_id)
group = self.sObj.getGroup(group_id)
member_dict = self.getGroupMemberIDs(group)
member_count = len(member_dict)
per_person_cost = round(cost / member_count, 2)
expense_members = []
print(per_person_cost * member_count, cost)
for member in member_dict:
expense_user = ExpenseUser()
expense_user.setId(member_dict[member])
expense_user.setFirstName(member)
expense_user.setOwedShare(str(per_person_cost))
if member == payer:
expense_user.setPaidShare(cost)
else:
expense_user.setPaidShare('0.00')
expense_members.append(expense_user)
if cost < per_person_cost * member_count:
remainder = (per_person_cost * float(member_count)) - cost
shuffle(expense_members)
i = 0
while remainder > 0.00:
owed = float(expense_members[i].getOwedShare())
owed -= 0.01
expense_members[i].setOwedShare(str(owed))
remainder -= 0.01
if i == member_count - 1:
i = 0
else:
i += 1
elif cost > per_person_cost * member_count:
remainder = round(cost - (per_person_cost * float(member_count)),
2)
print(remainder)
shuffle(expense_members)
i = 0
while remainder > 0.00:
owed = float(expense_members[i].getOwedShare())
owed += 0.01
expense_members[i].setOwedShare(str(owed))
remainder -= 0.01
if i == member_count - 1:
i = 0
else:
i += 1
expense.setUsers(expense_members)
expenses = self.sObj.createExpense(expense)
print('Successfully added to Splitwise. Expense ID:', expenses.getId())
def main():
if 'access_token' in session:
sObj.setAccessToken(session['access_token'])
user = sObj.getCurrentUser()
friends = sObj.getFriends()
friends_dict = {friend.getFirstName():friend.getId() for friend in friends}
groups = sObj.getGroups()
group_list = [group.getName() for group in groups]
group_id = [group.getId() for group in groups]
group_dict = {group.getName():group for group in groups}
with open(CSV_FILE_LOCATION, encoding='utf-8-sig') as data:
importedData = list(csv.DictReader(data))
# print(importedData)
##### Vet csv before proceeding #####
for i, row in enumerate(importedData):
if not (row.get('Debit') and row.get('Payer') and row.get('Group')):
return f"Remove blanks on row {i+1} in your CSV and try again."
with open(DB) as db_json:
db = json.load(db_json)
msg = []
# Begin loop reading the CSV
for row in importedData:
rowstring = f"{row.get('Date')}, {row.get('Debit')}, {row.get('Description')}, {row.get('Group')}, {row.get('Payer')}"
rsbytes = bytes(rowstring, 'utf-8')
hash_obj = hashlib.sha256(rsbytes)
hex_dig = hash_obj.hexdigest()
# print(hex_dig)
if hex_dig in db:
msg.append(f"{rowstring} looks like a duplicate, skipping")
continue
else:
db[hex_dig] = "True"
expense = Expense()
price = float(row['Debit'] or 0)
expense.setCost(price)
expense.setDate(row['Date'])
expense.setDescription(row['Description'])
expense.setGroupId(group_dict[row['Group']].getId())
members = group_dict[row['Group']].getMembers()
users = []
for member in members:
user = ExpenseUser()
user.setId(member.getId())
if member.getFirstName() == row['Payer']:
user.setPaidShare(price)
else:
user.setPaidShare(0)
users.append(user)
paid = 0
share = round(price/len(users),2)
for user in users:
user.setOwedShare(share)
paid = paid + share
diff = price - paid
if diff != 0:
user = random.choice(users)
user.setOwedShare(share + diff)
expense.setUsers(users)
expense = sObj.createExpense(expense)
msg.append('Import successful!')
with open(DB, 'w') as outfile:
json.dump(db, outfile)
msg.append('New entries recorded in db.')
message = "<br>".join(msg)
return message
return 'User is not logged in'
def add_group_expense(exp_data: dict):
"""
Utility function to add multiple expenses at once.
Pass in a dict of user_ids -> amount owed
"""
try:
total = float(exp_data['total'])
group_id = exp_data['group_id']
desc = exp_data['desc']
except KeyError:
print("total, group_id and desc are required fields in the expense")
return
expense = Expense()
expense.setCost(total)
expense.setDescription(desc)
expense.setGroupId(group_id)
sum = 0
for user_id, details in exp_data['users'].items():
try:
amt = float(details["amt"])
except KeyError:
print("'amt' is a required field")
return
except ValueError:
print(f"invalid amount {details['amt']}")
return
user = ExpenseUser()
user.setId(user_id)
user.setPaidShare(0)
user.setOwedShare(amt)
sum += amt
expense.addUser(user)
me = ExpenseUser()
me.setId(s.getCurrentUser().getId())
me.setPaidShare(total)
me.setOwedShare(total - sum)
expense.addUser(me)
if DRY_RUN:
print("--- dry run ---")
print_expense_details(expense)
return
resp, err = s.createExpense(expense)
# handle err
if err is not None:
print(err.getErrors())
return
print("Expense created!")
timestamp_comment = f"added at {datetime.datetime.now()} :)"
add_comment_to_expense(resp.getId(), timestamp_comment)
# extra comments
if "comment" in exp_data and exp_data["comment"] != "":
add_comment_to_expense(resp.getId(), comment)
def process(self, input):
app.logger.debug("Processing New Transaction Request")
output = BotConstants.NEW_EXPENSE_OUTPUT
splitwise_obj = SplitwiseProcessor.getSplitwiseObject(input)
current_user = splitwise_obj.getCurrentUser()
friends_list = splitwise_obj.getFriends()
user_list = []
split = SplitwiseProcessor.getInputFromRequest(
input, TransactionProcessor.SplitType.SPLIT, ErrorMessages.SPLIT,
True)
split = split.lower()
if split not in TransactionProcessor.SplitType.getSplitList():
raise BotException(ErrorMessages.WRONG_SPLIT)
name = SplitwiseProcessor.getInputFromRequest(input, BotConstants.NAME,
ErrorMessages.NAME, True)
currency = SplitwiseProcessor.getInputFromRequest(
input, BotConstants.CURRENCY)
description = SplitwiseProcessor.getInputFromRequest(
input, BotConstants.DESCRIPTION)
if description == "":
description = BotConstants.FROM_BOT
amount = 0
if BotConstants.AMOUNT in currency:
amount = currency[BotConstants.AMOUNT]
else:
amount = SplitwiseProcessor.getInputFromRequest(
input, BotConstants.AMOUNT, self.getAmountError(split), True)
group = str(
SplitwiseProcessor.getInputFromRequest(input, BotConstants.GROUP))
group_id = self.getGroupId(group, splitwise_obj.getGroups())
expense = Expense()
expense.setCost(amount)
if not group_id == -1:
expense.setGroupId(group_id)
mode = split.lower()
expense.setDescription(description)
# current user
paid, owed = self.getDistribution(mode, amount)
expense_user = self.getExpenseUser(current_user, paid, owed)
user_list.append(expense_user)
match = False
for friend in friends_list:
if friend.getFirstName().lower() == name.lower():
expense_user = self.getExpenseUser(friend, owed, paid)
match = True
if mode != TransactionProcessor.SplitType.PAID and mode != TransactionProcessor.SplitType.OWE:
expense_user.setPaidShare(str(0))
expense_user.setOwedShare(str(owed))
output += friend.getFirstName()
user_list.append(expense_user)
break
if not match:
raise BotException(ErrorMessages.NO_FRIEND.format(name=name))
expense.setUsers(user_list)
expense = splitwise_obj.createExpense(expense)
if expense.getId() is None:
raise BotException(ErrorMessages.GENERAL)
app.logger.debug("New Expense is created")
return output
