С того же времени в местной прессе начинают появляться его фельетоны, корреспонденции, театральные рецензии, статьи. Молодой автор сочиняет и юмористические рассказы, и одноактные пьесы для сельской художественной самодеятельности. В драматургии Завалишин пробует свои силы уже с начала 1920-х. Сохранилось его письмо к Мейерхольду 1921 г., в котором он пытается выяснить судьбу своей пьесы «Бывшие», переданной в ТЕО (где в то время служил Мейерхольд). «Она в Томске ставилась уж с рукописи и имела успех», — сообщает начинающий драматург (РГАЛИ. Ф. 998. Оп. 1. Ед. хр. 1570. Л. 1).
В июле 1921 г. Завалишин знакомится с Горьким, который в беседе с молодым литератором-мордвином рассказывает, что по материнской линии он тоже из мордвы. В следующем году молодого, активного и грамотного коммуниста отзывают в Москву — на работу в редакции газеты «Беднота», где он выступает в качестве фельетониста и очеркиста до начала 1930-х гг.
Первые сочинения Завалишина отданы пережитому во время Гражданской войны. Ранние пьесы: «Таежные гудки» — Сибирь, красное подполье времен колчаковщины; «Третий жених» — партизаны в Сибири, тоже во время Колчака; «Сплетня» — развал крестьянской семьи, борьба за новые отношения в семье. В архиве сохранились договор и переписка Госиздата с Завалишиным по поводу его пьес «Частное дело» (старая и новая семейная мораль в деревне) и «Перед концом». Последнюю автор оценивает как «исторический, идеологический и литературно-художественный ответ двум пьесам — „Бронепоезду 14–69“ Вс. Иванова и „Дни Турбиных“ М. Булгакова» (РГАЛИ. Ф. 611. Оп. 2. Ед. хр. 103. Л. 4,11–12. 12 дек. 1925 — 31 дек. 1929).
К середине 1920-х Завалишин вхож в столичные литературные круги, он член Общества крестьянских писателей (1927), а также — месткома писателей и московского Общества драматических писателей и композиторов (МОДПиК). На чистке писателей (осенью 1933 г.) сообщается:
«Начав писать, Завалишин неожиданно почувствовал тяжесть партийной дисциплины, его тянуло к пьесе, литературе, а партия бросала его на административную работу… <…> И тогда <…> чтобы иметь возможность писать, он вышел из партии. Ему казалось, что он может быть коммунистом, не будучи в партии. Завалишин ошибся. Скоро он почувствовал, что могучий источник его прежней силы, уверенности и ориентировки пропал <…> В результате — „Партбилет“, политически ошибочная пьеса. <…> И Завалишин сделал единственно правильный вывод: он вновь подал в партию. Писательская ячейка приняла его и — думается — не ошиблась».
(Острогорский Н., Абольников С. Дневник чистки // Лит. газ. 1933. 5 сент. № 41. С. 1)
«Партбилет», ставший вехой в литературной и человеческой судьбе драматурга, написан в 1928 г. Пьеса задумывалась как сатирическая бытовая комедия во время расцвета нэпа, а спектакль вышел к публике в месяцы великого перелома. В процессе многократных переделок «Партбилет» сменил жанр, превратившись в «трагикомедию». Комизм всячески приглушался, мотивировки поступков центрального героя, его предысторию (старый большевик — или эсер) автор переписывал на ходу, ломая логику характера. В центр пьесы вышел второстепенный персонаж, инвалид Гражданской войны и доморощенный мемуарист Шайкин. В критике промелькнуло сравнение персонажа Завалишина с зощенковскими героями, параллель, многое объясняющая в дальнейшей судьбе драматурга.
После снятия «Партбилета» Завалишин пишет производственную пьесу «Стройфронт!», посвященную строительству Магнитки, не оставляет дальнейших театральных планов. Пробует свои силы как прозаик и даже как литературовед — готовит к изданию сочинения Некрасова и Салтыкова-Щедрина для «Литературного наследства». Продолжает дружить с актером Дм. Орловым, воплотившим его Шайкина. В одном из писем к нему 30 июня 1934 г. из Коктебеля сообщает, что не живет, а только дышит, но надеется, что «блеск и синева моря пробудят <…> мои прежние искристые, засохшие теперь источники веселья, жизнерадостности» (Письма Завалишина А. И. Орлову Д. Н. // РГАЛИ. Ф. 2228. Оп. 1. Ед. хр. 3. Л. 1 об.).
До ареста Завалишину остается пять лет. «Умер в тюрьме», — сообщает современное издание (Между молотом и наковальней: Союз советских писателей СССР. Документы и материалы. М.: Росспэн. 2010. С. 252). Посмертно реабилитирован.