Skip to main content

Socket wrapper for asyncio

Project description

Flexi-Socket

Note: This project is under ongoing development, with additional features planned for future releases.

Overview

Flexi-Socket is a Python library for developing robust server and client socket applications. It supports TCP protocol and is capable of handling connections, messages, and disconnections in both server and client modes. The library is equipped with message classification and string message handling, suitable for a range of network programming needs.

View on PyPI

Installation

pip install flexi-socket

General Setup

Importing the Library

from flexi_socket import FlexiSocket, Connection, Protocol, Mode

Creating a Socket

Create a socket instance for server or client:

socket = FlexiSocket(mode=Mode.SERVER, protocol=Protocol.TCP, port=8081, read_buffer_size=1024)  # Server
# or
socket = FlexiSocket(mode=Mode.CLIENT, protocol=Protocol.TCP, host="0.0.0.0", port=8080)  # Client

Event Handlers

Implement event handlers for operations like connect, message handling, and disconnect:

On Connect

@socket.on_connect()
async def on_connect(connection: Connection):
    print(f"Connected to {connection}")
    # Additional logic

On Disconnect

@socket.on_disconnect()
async def on_disconnect(connection: Connection):
    print(f"Disconnected from {connection}")
    # Additional logic

On Message

@socket.on_message()
async def on_message(connection: Connection, message: str):
    print(f"Received message from {connection}: {message}")
    # Additional logic

After Receive and Before Send

This can be used to implement start/end bytes, checksums, message cleanup or any other logic that needs to be applied to the message but without cluster the on_message handler. A future implementation will combine these two handlers into one using something like yield.

@socket.after_receive()
async def after_receive(connection: Connection, message: str):
    print(f"Received message from {connection}: {message}")
    # Additional logic


@socket.before_send()
async def before_send(connection: Connection, message: str):
    print(f"Sending message to {connection}: {message}")
    # Additional logic

Sending Messages

To send a message:

await socket.send("Your message here")

Starting the Socket

socket.start()
# or 
await socket.start_async()

Using Message Classification

Message classification allows handling different message types based on first message received from a client. This is useful because a lot of protocols include a handshake message, which can be used to identify the type of client (very popular in alarm systems).

This is an attempt to make something similar to http routing, but for classic socket programming.

class ClientTypes(ClientClassifier):
    client_types = ["001", "002", "003"]

    def classify(self, first_message):
        if first_message.startswith("!!"):
            return ClientTypes.client_types[0]
        elif first_message.startswith("##"):
            return ClientTypes.client_types[1]
        else:
            return ClientClassifier.DEFAULT

Using the Classifier

@socket.on_message(ClientTypes.client_types[0])
async def handle_client_001(client: Connection, message: str):
    if client.is_first_message_from_client:
        print("First message from client")
    print(f"Client {client} sent {message}")
    await client.send("Hello from server! You are type 001")

Planned Features

  • UDP Protocol Support: Future implementation to handle UDP connections.
  • Binary Message Support: Upgrade to allow sending and receiving binary data, in addition to string messages.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

flexi-socket-0.0.8.tar.gz (7.4 kB view details)

Uploaded Source

Built Distribution

flexi_socket-0.0.8-py3-none-any.whl (7.5 kB view details)

Uploaded Python 3

File details

Details for the file flexi-socket-0.0.8.tar.gz.

File metadata

  • Download URL: flexi-socket-0.0.8.tar.gz
  • Upload date:
  • Size: 7.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.18

File hashes

Hashes for flexi-socket-0.0.8.tar.gz
Algorithm Hash digest
SHA256 c82cb4bb36ed8f5068705a144c16b3f57564f66a455b89f8a57d8358372dd7ae
MD5 93fcaf0974360dcc9ffb86f1ba7f4cd1
BLAKE2b-256 78894bcf7c946af520762ca82f67aef8d13519ae7bc3285450b912064fb55ccd

See more details on using hashes here.

File details

Details for the file flexi_socket-0.0.8-py3-none-any.whl.

File metadata

File hashes

Hashes for flexi_socket-0.0.8-py3-none-any.whl
Algorithm Hash digest
SHA256 804ec3af6b947fd97834980058c60680318d795d80c1203c041553a1650a97ad
MD5 cb733432a15fee2f1094870327ec78fe
BLAKE2b-256 e287a986aaad207c196d82798d145a39609c82d50defbbe4d3e33073f83ca12b

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page