From: Guido van Rossum <guido@python.org>
Date: Wed, 12 Jun 1996 04:24:52 +0000 (+0000)
Subject: added capitalize()
X-Git-Tag: v1.4b1~66
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=2745753b91b1f83f8c24a2e3ed39f103ba1a1043;p=python

added capitalize()
---

diff --git a/Modules/stropmodule.c b/Modules/stropmodule.c
index a82c3138ed..52d4cee366 100644
--- a/Modules/stropmodule.c
+++ b/Modules/stropmodule.c
@@ -367,6 +367,50 @@ strop_upper(self, args)
 }
 
 
+static object *
+strop_capitalize(self, args)
+	object *self; /* Not used */
+	object *args;
+{
+	char *s, *s_new;
+	int i, n;
+	object *new;
+	int changed;
+
+	if (!getargs(args, "s#", &s, &n))
+		return NULL;
+	new = newsizedstringobject(NULL, n);
+	if (new == NULL)
+		return NULL;
+	s_new = getstringvalue(new);
+	changed = 0;
+	{
+		int c = Py_CHARMASK(*s++);
+		if (islower(c)) {
+			changed = 1;
+			*s_new = toupper(c);
+		} else
+			*s_new = c;
+		s_new++;
+	}
+	for (i = 1; i < n; i++) {
+		int c = Py_CHARMASK(*s++);
+		if (isupper(c)) {
+			changed = 1;
+			*s_new = tolower(c);
+		} else
+			*s_new = c;
+		s_new++;
+	}
+	if (!changed) {
+		DECREF(new);
+		INCREF(args);
+		return args;
+	}
+	return new;
+}
+
+
 static object *
 strop_swapcase(self, args)
 	object *self; /* Not used */
@@ -538,6 +582,7 @@ static struct methodlist strop_methods[] = {
 	{"atof",	strop_atof},
 	{"atoi",	strop_atoi},
 	{"atol",	strop_atol},
+	{"capitalize",	strop_capitalize},
 	{"find",	strop_find},
 	{"join",	strop_joinfields, 1},
 	{"joinfields",	strop_joinfields, 1},