Coverage for polar/enums.py: 59%

1from datetime import datetime 2 ctx1ab

2from enum import StrEnum 2 ctx1ab

3from typing import Literal 2 ctx1ab

4 

5from dateutil.relativedelta import relativedelta 2 ctx1ab

6 

7 

8class Platforms(StrEnum): 2 ctx1ab

9 github = "github" 2 ctx1ab

10 

11 

12class PaymentProcessor(StrEnum): 2 ctx1ab

13 stripe = "stripe" 2 ctx1ab

14 

15 

16class AccountType(StrEnum): 2 ctx1ab

17 stripe = "stripe" 2 ctx1ab

18 manual = "manual" 2 ctx1ab

19 open_collective = "open_collective" 2 ctx1ab

20 

21 def get_display_name(self) -> str: 2 ctx1ab

22 return { 

23 AccountType.stripe: "Stripe Connect Express", 

24 AccountType.open_collective: "Open Collective", 

25 AccountType.manual: "Manual", 

26 }[self] 

27 

28 

29class SubscriptionRecurringInterval(StrEnum): 2 ctx1ab

30 day = "day" 2 ctx1ab

31 week = "week" 2 ctx1ab

32 month = "month" 2 ctx1ab

33 year = "year" 2 ctx1ab

34 

35 def as_literal(self) -> Literal["day", "week", "month", "year"]: 2 ctx1ab

36 return self.value 

37 

38 def get_next_period(self, d: datetime, leap: int = 1) -> datetime: 2 ctx1ab

39 match self: 

40 case SubscriptionRecurringInterval.day: 

41 return d + relativedelta(days=leap) 

42 case SubscriptionRecurringInterval.week: 

43 return d + relativedelta(weeks=leap) 

44 case SubscriptionRecurringInterval.month: 

45 return d + relativedelta(months=leap) 

46 case SubscriptionRecurringInterval.year: 

47 return d + relativedelta(years=leap) 

48 

49 

50class SubscriptionProrationBehavior(StrEnum): 2 ctx1ab

51 invoice = "invoice" # Invoice immediately 2 ctx1ab

52 prorate = "prorate" # Add prorations to next invoice 2 ctx1ab

53 

54 def to_stripe(self) -> Literal["always_invoice", "create_prorations"]: 2 ctx1ab

55 if self == SubscriptionProrationBehavior.invoice: 

56 return "always_invoice" 

57 if self == SubscriptionProrationBehavior.prorate: 

58 return "create_prorations" 

59 raise ValueError(f"Invalid proration behavior: {self}") 

60 

61 

62class InvoiceNumbering(StrEnum): 2 ctx1ab

63 organization = "organization" 2 ctx1ab

64 customer = "customer" 2 ctx1ab

65 

66 

67class TokenType(StrEnum): 2 ctx1ab

68 client_secret = "polar_client_secret" 2 ctx1ab

69 client_registration_token = "polar_client_registration_token" 2 ctx1ab

70 authorization_code = "polar_authorization_code" 2 ctx1ab

71 access_token = "polar_access_token" 2 ctx1ab

72 refresh_token = "polar_refresh_token" 2 ctx1ab

73 personal_access_token = "polar_personal_access_token" 2 ctx1ab

74 organization_access_token = "polar_organization_access_token" 2 ctx1ab

75 customer_session_token = "polar_customer_session_token" 2 ctx1ab

76 user_session_token = "polar_user_session_token" 2 ctx1ab

77 

78 

79class RateLimitGroup(StrEnum): 2 ctx1ab

80 web = "web" 2 ctx1ab

81 default = "default" 2 ctx1ab

82 elevated = "elevated" 2 ctx1ab