Update app.py

app.py

@@ -12,22 +12,13 @@ def machine_id(id):
     return (id >> 12) & 0b1111111111
 
 
-def server_id(id):
-    return (id >> 12) & 0b11111
-
-
-def datacenter_id(id):
-    return (id >> 17) & 0b11111
-
-
 def creation_time(id):
     return id >> 22
 
 
 def snow_to_components(snowflake_id: int):
     timestamp_ms = creation_time(snowflake_id) + SNOWFLAKE_EPOCH
-    datacenter_id_val = datacenter_id(snowflake_id)
-    server_id_val = server_id(snowflake_id)
+    # datacenter_id_val = datacenter_id(snowflake_id)
     machine_id_val = machine_id(snowflake_id)
     sequence_id_val = sequence_id(snowflake_id)
     readable_time = time.strftime(
@@ -36,8 +27,6 @@ def snow_to_components(snowflake_id: int):
     return (
         readable_time,
         timestamp_ms,
-        datacenter_id_val,
-        server_id_val,
         machine_id_val,
         sequence_id_val,
     )
@@ -45,49 +34,30 @@ def snow_to_components(snowflake_id: int):
 
 def components_to_snow(
     timestamp_ms: int,
-    datacenter_id_val: int,
-    server_id_val: int,
     machine_id_val: int,
     sequence_id_val: int,
 ):
-    timestamp_bits = bin(int(timestamp_ms) - SNOWFLAKE_EPOCH)[2:].zfill(41)
-    datacenter_id_bits = bin(datacenter_id_val)[2:].zfill(5)
-    server_id_bits = bin(server_id_val)[2:].zfill(5)
-    machine_id_bits = bin(machine_id_val)[2:].zfill(10)
-    sequence_bits = bin(sequence_id_val)[2:].zfill(12)
-    binary_snowflake = (
-        timestamp_bits
-        + machine_id_bits
-        + server_id_bits
-        + datacenter_id_bits
-        + sequence_bits
-    )
-    snowflake_id = int(binary_snowflake, 2)
-    return snowflake_id
+    # timestamp_bits = bin()[2:].zfill(41)
+    time_component = bin(int(timestamp_ms) - SNOWFLAKE_EPOCH)
+    return int(time_component[2:].zfill(42) + bin(machine_id_val)[2:].zfill(10) + bin(sequence_id_val)[2:].zfill(12),2)
 
 
 def update_components(snowflake_input_str):
     if snowflake_input_str:
         snowflake_input = int(snowflake_input_str)
         return snow_to_components(snowflake_input)
-    return None, None, None, None, None, None
+    return None, None, None, None, None
 
 
 def update_snowflake(
     _,
     timestamp_input,
-    datacenter_input,
-    server_input,
     machine_input,
     sequence_input,
 ):
-    if all(
-        [timestamp_input, datacenter_input, server_input, machine_input, sequence_input]
-    ):
+    if all([timestamp_input, machine_input, sequence_input]):
         snowflake_id = components_to_snow(
             int(timestamp_input),
-            int(datacenter_input),
-            int(server_input),
             int(machine_input),
             int(sequence_input),
         )
@@ -101,21 +71,17 @@ with gr.Blocks() as demo:
         snowflake = gr.Textbox(label="Snowflake ID", interactive=True)
     with gr.Accordion("Snowflake ID Components", open=False):
         readable_time = gr.Textbox(label="Timestamp (Readable)", interactive=False)
-        timestamp = gr.Number(label="Timestamp (ms)", interactive=True)
-        datacenter = gr.Number(0, label="Datacenter ID", step=1, interactive=True)
-        server = gr.Number(0, label="Server ID", step=1, interactive=True)
+        timestamp = gr.Number(0,label="Timestamp (ms)", interactive=True)
         machine = gr.Number(0, label="Machine ID", step=1, interactive=True)
         sequence = gr.Number(0, label="Sequence ID", step=1, interactive=True)
         id_components = [
             readable_time,
             timestamp,
-            datacenter,
-            server,
             machine,
             sequence,
         ]
     snowflake.input(update_components, inputs=snowflake, outputs=id_components)
-    for component in [timestamp, datacenter, server, machine, sequence]:
+    for component in [timestamp, machine, sequence]:
         component.input(update_snowflake, inputs=id_components, outputs=snowflake)
 
-demo.launch()
+demo.launch()