You've already forked system-design-primer
mirror of
https://github.com/donnemartin/system-design-primer.git
synced 2025-06-12 21:47:33 +02:00
Add missing enum imports (#157)
This commit is contained in:
@ -1,4 +1,5 @@
|
|||||||
from abc import ABCMeta
|
from abc import ABCMeta
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
class UserService(object):
|
class UserService(object):
|
||||||
|
@ -1,12 +1,16 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
class DefaultCategories(Enum):
|
class DefaultCategories(Enum):
|
||||||
|
|
||||||
HOUSING = 0
|
HOUSING = 0
|
||||||
FOOD = 1
|
FOOD = 1
|
||||||
GAS = 2
|
GAS = 2
|
||||||
SHOPPING = 3
|
SHOPPING = 3
|
||||||
...
|
# ...
|
||||||
|
|
||||||
|
|
||||||
seller_category_map = {}
|
seller_category_map = {}
|
||||||
seller_category_map['Exxon'] = DefaultCategories.GAS
|
seller_category_map['Exxon'] = DefaultCategories.GAS
|
||||||
@ -44,4 +48,3 @@ class Budget(object):
|
|||||||
|
|
||||||
def override_category_budget(self, category, amount):
|
def override_category_budget(self, category, amount):
|
||||||
self.categories_to_budget_map[category] = amount
|
self.categories_to_budget_map[category] = amount
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user