summaryrefslogtreecommitdiffstats
path: root/Objects/moduleobject.c
blob: 7b9e0e919f989480381718a87db7d8aa45b949e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* Module object implementation */

#include <stdio.h>

#include "PROTO.h"
#include "object.h"
#include "stringobject.h"
#include "dictobject.h"
#include "moduleobject.h"
#include "objimpl.h"
#include "errors.h"

typedef struct {
	OB_HEAD
	object *md_name;
	object *md_dict;
} moduleobject;

object *
newmoduleobject(name)
	char *name;
{
	moduleobject *m = NEWOBJ(moduleobject, &Moduletype);
	if (m == NULL)
		return NULL;
	m->md_name = newstringobject(name);
	m->md_dict = newdictobject();
	if (m->md_name == NULL || m->md_dict == NULL) {
		DECREF(m);
		return NULL;
	}
	return (object *)m;
}

object *
getmoduledict(m)
	object *m;
{
	if (!is_moduleobject(m)) {
		errno = EBADF;
		return NULL;
	}
	return ((moduleobject *)m) -> md_dict;
}

int
setmoduledict(m, v)
	object *m;
	object *v;
{
	if (!is_moduleobject(m))
		return errno = EBADF;
	if (!is_dictobject(v))
		return errno = EINVAL;
	DECREF(((moduleobject *)m) -> md_dict);
	INCREF(v);
	((moduleobject *)m) -> md_dict = v;
	return 0;
}

/* Methods */

static void
moduledealloc(m)
	moduleobject *m;
{
	if (m->md_name != NULL)
		DECREF(m->md_name);
	if (m->md_dict != NULL)
		DECREF(m->md_dict);
	free((char *)m);
}

static void
moduleprint(m, fp, flags)
	moduleobject *m;
	FILE *fp;
	int flags;
{
	fprintf(fp, "<module %s>", getstringvalue(m->md_name));
}

static object *
modulerepr(m)
	moduleobject *m;
{
	char buf[100];
	sprintf(buf, "<module %.80s>", getstringvalue(m->md_name));
	return newstringobject(buf);
}

static object *
modulegetattr(m, name)
	moduleobject *m;
	char *name;
{
	object *res = dictlookup(m->md_dict, name);
	if (res == NULL)
		err_setstr(NameError, name);
	else
		INCREF(res);
	return res;
}

static int
modulesetattr(m, name, v)
	moduleobject *m;
	char *name;
	object *v;
{
	if (v == NULL)
		return dictremove(m->md_dict, name);
	else
		return dictinsert(m->md_dict, name, v);
}

typeobject Moduletype = {
	OB_HEAD_INIT(&Typetype)
	0,			/*ob_size*/
	"module",		/*tp_name*/
	sizeof(moduleobject),	/*tp_size*/
	0,			/*tp_itemsize*/
	moduledealloc,	/*tp_dealloc*/
	moduleprint,	/*tp_print*/
	modulegetattr,	/*tp_getattr*/
	modulesetattr,	/*tp_setattr*/
	0,		/*tp_compare*/
	modulerepr,	/*tp_repr*/
};