mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2025-08-09 13:29:47 +00:00
Change the way connection modes are specified
This commit is contained in:
22
telethon/_misc/enums.py
Normal file
22
telethon/_misc/enums.py
Normal file
@@ -0,0 +1,22 @@
|
||||
from enum import Enum
|
||||
|
||||
|
||||
class ConnectionMode(Enum):
|
||||
FULL = 'full'
|
||||
INTERMEDIATE = 'intermediate'
|
||||
ABRIDGED = 'abridged'
|
||||
OBFUSCATED = 'obfuscated'
|
||||
HTTP = 'http'
|
||||
|
||||
|
||||
def parse_conn_mode(mode):
|
||||
if isinstance(mode, ConnectionMode):
|
||||
return mode
|
||||
elif isinstance(mode, str):
|
||||
for cm in ConnectionMode:
|
||||
if mode == cm.value:
|
||||
return cm
|
||||
|
||||
raise ValueError(f'unknown connection mode: {mode!r}')
|
||||
else:
|
||||
raise TypeError(f'not a valid connection mode: {type(mode).__name__!r}')
|
Reference in New Issue
Block a user