vardef 0.1.0

from vardef import vardef


vars_defined = 0

@vardef
def somevar() -> int:
    global vars_defined
    vars_defined += 1
    return 42

@vardef
def othervar() -> int:
    global vars_defined
    vars_defined += 1
    return 73

print(vars_defined)
# 2

print(somevar)
# 42

print(othervar)
# 73

print(type(somevar))
# <class 'int'>

print(type(othervar))
# <class 'int'>

from unittest.mock import Mock
from vardef import vardef

from somewhere import User


@vardef
def user_with_read_role() -> Mock:
    user_mock = Mock(User)
    user_mock.roles = ["READ"]
    return user_mock

print(user_with_read_role.roles)
# ['READ']

print(type(user_with_read_role))
# <class 'unittest.mock.Mock'>

import pandas as pd

from butterfree.extract import Source
from butterfree.extract.readers import TableReader
from butterfree.clients import SparkClient
from vardef import vardef


spark_client = SparkClient()

@vardef
def df() -> pd.DataFrame:
    source = Source(
        readers=[TableReader(
            id="colors",
            database="datalake_colors",
            table="colors",
        )],
        query="""
            SELECT * FROM colors
        """,
    )
    return source.construct(spark_client)

df.createOrReplaceTempView("colors")
...

from vardef import vardef


@vardef
def buggy() -> int:
    return 4 / 0

# Traceback (most recent call last):
#   File "./buggy.py", line 5, in <module>
#     def buggy() -> int:
#   File "./vardef/__init__.py", line 7, in vardef
#     return define_var()
#   File "./buggy.py", line 6, in buggy
#     return 4 / 0
# ZeroDivisionError: division by zero