stream settings: Refactor out STREAM_TRAFFIC_CALCULATION_MIN_AGE_DAYS.

This commit is contained in:
Rishi Gupta
2018-06-20 12:08:55 -07:00
committed by Tim Abbott
parent c6048527bf
commit f63bcb99d7

View File

@@ -3901,6 +3901,8 @@ def get_streams_traffic(streams: Optional[Iterable[Stream]]=None) -> Dict[int, i
def round_to_2_significant_digits(number: int) -> int:
return int(round(number, 2 - len(str(number))))
STREAM_TRAFFIC_CALCULATION_MIN_AGE_DAYS = 7
def get_average_weekly_stream_traffic(stream_id: int, stream_date_created: datetime.datetime,
recent_traffic: QuerySet) -> int:
try:
@@ -3912,7 +3914,7 @@ def get_average_weekly_stream_traffic(stream_id: int, stream_date_created: datet
if stream_age >= 28:
average_weekly_traffic = int(stream_traffic // 4)
elif stream_age >= 7:
elif stream_age >= STREAM_TRAFFIC_CALCULATION_MIN_AGE_DAYS:
average_weekly_traffic = int(stream_traffic * 7 // stream_age)
else:
average_weekly_traffic = stream_traffic
@@ -3920,7 +3922,8 @@ def get_average_weekly_stream_traffic(stream_id: int, stream_date_created: datet
return round_to_2_significant_digits(average_weekly_traffic)
def is_old_stream(stream_date_created: datetime.datetime) -> bool:
return (datetime.date.today() - stream_date_created.date()).days >= 7
return (datetime.date.today() - stream_date_created.date()).days \
>= STREAM_TRAFFIC_CALCULATION_MIN_AGE_DAYS
def encode_email_address(stream: Stream) -> str:
return encode_email_address_helper(stream.name, stream.email_token)