156 lines
2.6 KiB
C
156 lines
2.6 KiB
C
/* 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)) {
|
|
err_badarg();
|
|
return NULL;
|
|
}
|
|
return ((moduleobject *)m) -> md_dict;
|
|
}
|
|
|
|
int
|
|
setmoduledict(m, v)
|
|
object *m;
|
|
object *v;
|
|
{
|
|
if (!is_moduleobject(m)) {
|
|
err_badarg();
|
|
return -1;
|
|
}
|
|
if (!is_dictobject(v)) {
|
|
err_badarg();
|
|
return -1;
|
|
}
|
|
DECREF(((moduleobject *)m) -> md_dict);
|
|
INCREF(v);
|
|
((moduleobject *)m) -> md_dict = v;
|
|
return 0;
|
|
}
|
|
|
|
char *
|
|
getmodulename(m)
|
|
object *m;
|
|
{
|
|
if (!is_moduleobject(m)) {
|
|
err_badarg();
|
|
return NULL;
|
|
}
|
|
return getstringvalue(((moduleobject *)m) -> md_name);
|
|
}
|
|
|
|
/* 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;
|
|
if (strcmp(name, "__dict") == 0) {
|
|
INCREF(m->md_dict);
|
|
return m->md_dict;
|
|
}
|
|
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 (strcmp(name, "__dict") == 0) {
|
|
/* Can't allow assignment to __dict, it would screw up
|
|
module's functions which still use the old dictionary. */
|
|
err_setstr(NameError, "__dict is a reserved member name");
|
|
return NULL;
|
|
}
|
|
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*/
|
|
};
|