Skip to content

Commit

Permalink
Update streams.py
Browse files Browse the repository at this point in the history
  • Loading branch information
DLT1412 committed Jun 19, 2024
1 parent 363c4b1 commit dac0832
Showing 1 changed file with 34 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,10 @@
from abc import ABC, abstractmethod
from datetime import datetime
from decimal import Decimal
from enum import Enum
if sys.version_info >= (3, 11):
from enum import StrEnum
else:
from enum import Enum
from functools import total_ordering
from typing import Any, Dict, Iterable, List, Mapping, MutableMapping, Optional, Tuple, TypeVar, Union

Expand Down Expand Up @@ -141,21 +144,39 @@ def __lt__(self, other):
return self.current_stream_state < other


class ReportLevel(str, Enum):
ADVERTISER = "ADVERTISER"
CAMPAIGN = "CAMPAIGN"
ADGROUP = "ADGROUP"
AD = "AD"
if sys.version_info >= (3, 11):

class ReportLevel(StrEnum):
ADVERTISER = "ADVERTISER"
CAMPAIGN = "CAMPAIGN"
ADGROUP = "ADGROUP"
AD = "AD"

class ReportGranularity(str, Enum):
LIFETIME = "LIFETIME"
DAY = "DAY"
HOUR = "HOUR"
class ReportGranularity(StrEnum):
LIFETIME = "LIFETIME"
DAY = "DAY"
HOUR = "HOUR"

@classmethod
def default(cls):
return cls.DAY
@classmethod
def default(cls):
return cls.DAY

else:

class ReportLevel(str, Enum):
ADVERTISER = "ADVERTISER"
CAMPAIGN = "CAMPAIGN"
ADGROUP = "ADGROUP"
AD = "AD"

class ReportGranularity(str, Enum):
LIFETIME = "LIFETIME"
DAY = "DAY"
HOUR = "HOUR"

@classmethod
def default(cls):
return cls.DAY


class Hourly:
Expand Down

0 comments on commit dac0832

Please sign in to comment.